File size: 851 Bytes
a6ebd25
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
27
28
29
Logging training
Running DummyRegressor()
r2: -0.001 neg_mean_squared_error: -7523.699
=== new best DummyRegressor() (using r2):
r2: -0.001 neg_mean_squared_error: -7523.699

Running DecisionTreeRegressor(max_depth=1)
r2: 0.800 neg_mean_squared_error: -1504.400
=== new best DecisionTreeRegressor(max_depth=1) (using r2):
r2: 0.800 neg_mean_squared_error: -1504.400

Running DecisionTreeRegressor(max_depth=5)
r2: 0.999 neg_mean_squared_error: -6.230
=== new best DecisionTreeRegressor(max_depth=5) (using r2):
r2: 0.999 neg_mean_squared_error: -6.230

Running Ridge(alpha=10)
r2: 1.000 neg_mean_squared_error: -1.068
=== new best Ridge(alpha=10) (using r2):
r2: 1.000 neg_mean_squared_error: -1.068

Running Lasso(alpha=10)
r2: 0.986 neg_mean_squared_error: -101.572

Best model:
Ridge(alpha=10)
Best Scores:
r2: 1.000 neg_mean_squared_error: -1.068