Diamond Price Prediction and Visualization

Alican AKCA
Jan 23, 2021

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.

--

--

Alican AKCA

Math Student at Izmir University of Economics. I have been working on Machine Learning and AI. alicanakca.space