From d92ed343fef98c4cfbf2086b6f4b1a5477da8fca Mon Sep 17 00:00:00 2001 From: schneefux Date: Sat, 20 Feb 2016 19:42:56 +0100 Subject: linear regression, smaller improvements --- linear.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 linear.py (limited to 'linear.py') diff --git a/linear.py b/linear.py new file mode 100644 index 0000000..36ef411 --- /dev/null +++ b/linear.py @@ -0,0 +1,22 @@ +#!/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) +squares=[] +for n in range(len(p)): + squares.append((y_test[n] - p[n])**2) + +print(np.mean(squares)) -- cgit v1.3.1