summaryrefslogtreecommitdiff
path: root/linear.py
diff options
context:
space:
mode:
Diffstat (limited to 'linear.py')
-rw-r--r--linear.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/linear.py b/linear.py
index 36ef411..f6d1568 100644
--- a/linear.py
+++ b/linear.py
@@ -15,6 +15,10 @@ 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)