Best model | name | model_type | metric_type | metric_value | train_time |
---|---|---|---|---|---|
1_Baseline | Baseline | rmse | 0.519978 | 0.2 | |
2_DecisionTree | Decision Tree | rmse | 0.454812 | 10.11 | |
3_Linear | Linear | rmse | 0.424076 | 7.83 | |
4_Default_Xgboost | Xgboost | rmse | 0.28695 | 8.5 | |
5_Default_NeuralNetwork | Neural Network | rmse | 0.383513 | 3.28 | |
6_Default_RandomForest | Random Forest | rmse | 0.400469 | 22.63 | |
4_Default_Xgboost_categorical_mix | Xgboost | rmse | 0.281327 | 9.91 | |
the best | Ensemble | Ensemble | rmse | 0.280161 | 0.16 |