blob: f6d1568c060904979d14cc7c2ba2a8a936a28a89 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#!/usr/bin/python3
from sklearn.datasets import *
from sklearn.linear_model import LinearRegression
from sklearn.cross_validation import train_test_split
import numpy as np
SEED = 42
np.random.seed(SEED)
data = load_boston()
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, train_size=0.85)
model = LinearRegression()
model.fit(data.data, data.target)
p=model.predict(X_test)
print(p)
print('diff')
print('')
print(y_test - p)
squares=[]
for n in range(len(p)):
squares.append((y_test[n] - p[n])**2)
print(np.mean(squares))
|
