Price Prediction Of Diamonds

Subscribe to my newsletter and never miss my upcoming articles

Listen to this article

I used Decision Tree, Random Forest, Linear, K-Neighbors and XGBoost Regressions with Python. I chose the best model with the Cross-Validation method. I used the R-Squared method to check that it gives a consistent result. As a result predicted with nearly 99% success rate. There is no 'Overfitting' problem.

Graph.png

I used the 10-Layer Cross-Validation method. The results are below:

2.png

It is XGBoost that works best in these datas. I used this regression model for prediction.

No Comments Yet