forked from pieroit/corso_ml_python_youtube_pollo
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathhello.py
More file actions
28 lines (18 loc) · 652 Bytes
/
Copy pathhello.py
File metadata and controls
28 lines (18 loc) · 652 Bytes
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
from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_absolute_error
from sklearn.model_selection import train_test_split
import numpy as np
np.random.seed(2)
dataset = load_boston()
X = dataset['data']
y = dataset['target']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
model = LinearRegression()
model.fit(X_train, y_train)
p_train = model.predict(X_train)
p_test = model.predict(X_test)
mae_train = mean_absolute_error(y_train, p_train)
print('MAE train', mae_train)
mae_test = mean_absolute_error(y_test, p_test)
print('MAE test', mae_test)