Best model | name | model_type | metric_type | metric_value | train_time |
---|---|---|---|---|---|
1_Baseline | Baseline | rmse | 15.6811 | 0.19 | |
2_DecisionTree | Decision Tree | rmse | 12.3305 | 10.12 | |
3_Linear | Linear | rmse | 17.3341 | 2.57 | |
4_Default_Xgboost | Xgboost | rmse | 6.97197 | 4.47 | |
5_Default_NeuralNetwork | Neural Network | rmse | 10.7188 | 0.79 | |
6_Default_RandomForest | Random Forest | rmse | 13.2731 | 3.08 | |
the best | Ensemble | Ensemble | rmse | 6.29788 | 0.11 |