Best model | name | model_type | metric_type | metric_value | train_time |
---|---|---|---|---|---|
1_Baseline | Baseline | rmse | 65.581 | 0.21 | |
2_DecisionTree | Decision Tree | rmse | 52.1432 | 12.51 | |
3_Linear | Linear | rmse | 39.6384 | 6.33 | |
4_Default_Xgboost | Xgboost | rmse | 42.7449 | 8.92 | |
5_Default_NeuralNetwork | Neural Network | rmse | 43.9206 | 2.71 | |
6_Default_RandomForest | Random Forest | rmse | 45.7597 | 18.23 | |
4_Default_Xgboost_categorical_mix | Xgboost | rmse | 43.2199 | 8.65 | |
the best | Ensemble | Ensemble | rmse | 38.974 | 0.14 |