Diamond Price Prediction and Visualization

Hi! 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 i think.

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

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

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store