Best model | name | model_type | metric_type | metric_value | train_time |
---|---|---|---|---|---|
1_Baseline | Baseline | rmse | 39.9364 | 0.29 | |
2_DecisionTree | Decision Tree | rmse | 8.12996 | 20.69 | |
3_Linear | Linear | rmse | 5.8666 | 9.68 | |
the best | 4_Default_Xgboost | Xgboost | rmse | 1.4182 | 94.3 |
5_Default_NeuralNetwork | Neural Network | rmse | 11.1378 | 6.79 | |
6_Default_RandomForest | Random Forest | rmse | 6.90049 | 23.75 | |
Ensemble | Ensemble | rmse | 1.4182 | 0.11 |