srg's picture
Upload ./ with huggingface_hub
a6ebd25
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