Abstract:This study conducted a comprehensive analysis of multiple supervised machine learning models, regressors and classifiers, to accurately predict diamond prices. Diamond pricing is a complex task due to the non-linear relationships between key features such as carat, cut, clarity, table, and depth. The analysis aimed to develop an accurate predictive model by utilizing both regression and classification approaches. To preprocess the data, the study employed various techniques. The work addressed outliers, standardized the predictors, performed median imputation of missing values, and resolved multicollinearity issues. Equal-width binning on the cut variable was performed to handle class imbalance. Correlation-based feature selection was utilized to eliminate highly correlated variables, ensuring that only relevant features were included in the models. Outliers were handled using the inter-quartile range method, and numerical features were normalized through standardization. Missing values in numerical features were imputed using the median, preserving the integrity of the dataset. Among the models evaluated, the RF regressor exhibited exceptional performance. It achieved the lowest root mean squared error (RMSE) of 523.50, indicating superior accuracy compared to the other models. The RF regressor also obtained a high R-squared ( ) score of 0.985, suggesting it explained a significant portion of the variance in diamond prices. Furthermore, the area under the curve with RF classifier for the test set was 1.00 , indicating perfect classification performance. These results solidify the RF's position as the best-performing model in terms of accuracy and predictive power, both in regression and classification. The MLP regressor showed promising results with an RMSE of 563.74 and an score of 0.980, demonstrating its ability to capture the complex relationships in the data. Although it achieved slightly higher errors than the RF regressor, further analysis is needed to determine its suitability and potential advantages compared to the RF regressor. The XGBoost Regressor achieved an RMSE of 612.88 and an score of 0.972, indicating its effectiveness in predicting diamond prices but with slightly higher errors compared to the RF regressor. The Boosted Decision Tree Regressor had an RMSE of 711.31 and an score of 0.968, demonstrating its ability to capture some of the underlying patterns but with higher errors than the RF and XGBoost models. In contrast, the KNN regressor yielded a higher RMSE of 1346.65 and a lower score of 0.887, indicating its inferior performance in accurately predicting diamond prices compared to the other models. Similarly, the Linear Regression model performed similarly to the KNN regressor, with an RMSE of 1395.41 and an score of 0.876. The Support Vector Regression model showed the highest RMSE of 3044.49 and the lowest score of 0.421, indicating its limited effectiveness in capturing the complex relationships in the data. Overall, the study demonstrates that the RF outperforms the other models in terms of accuracy and predictive power, as evidenced by its lowest RMSE, highest score, and perfect classification performance. This highlights its suitability for accurately predicting diamond prices. The study not only provides an effective tool for the diamond industry but also emphasizes the importance of considering both regression and classification approaches in developing accurate predictive models. The findings contribute valuable insights for pricing strategies, market trends, and decision-making processes in the diamond industry and related fields.

Incorporating a machine learning approach for commodity price prediction to generate a cut-off grade optimisation policy under grade uncertainty

A new optimisation model for the selection of optimal cut-off grade using multi-sequential decision algorithm in open-pit metalliferous deposits

Comparative study of multiple machine learning algorithms for risk level prediction in goaf

Cutoff Grade Optimization on Operative Decision Variables with Geological Uncertainty in an Underground Gold Mine: A Real Case Study

Data-Driven Robust Optimization Under Correlated Uncertainty: A Case Study Of Production Scheduling In Ethylene Plant (Reprinted From Computers And Chemical Engineering, Vol 109, Pg 48-67, 2017)

Simulation-based decision-making system for optimal mine production plan selection

Mid-long term boundary dynamic optimization of open-pit coal mine considering coal price fluctuation

Research on Decision Optimization and the Risk Measurement of the Power Generation Side Based on Quantile Data-Driven IGDT

A Method for Assessing the Impact of Changes in Demand for Coal on the Structure of Coal Grades Produced by Mines

Prediction of blast-induced air overpressure using a hybrid machine learning model and gene expression programming (GEP): A case study from an iron ore mine

Boundary optimization of inclined coal seam open-pit mine based on the ISSA–LSSVR coal price prediction method

Multi-step Metal Prices Forecasting Based on a Data Preprocessing Method and an Optimized Extreme Learning Machine by Marine Predators Algorithm

Modeling and forecasting of coal price based on influencing factors and time series

Integrating geometallurgy in the simulation of processing plant responses and applying policy gradient for shovel allocation in operating mining complexes

Forecasting carbon emissions future prices using the machine learning methods

Refinery Planning Optimization Based on Smart Predict-then-optimize Method under Exogenous Price Uncertainty

Incorporation of geometallurgical modelling into long-term production planning

A Hybrid Traditional and Machine Learning-Based Stacking-Based Ensemble Forecasting Approach for Coal Price Prediction

Optimisation of life-of-mine production scheduling for block-caving mines under mineral resource and material mixing uncertainty

Assessing predictive performance of supervised machine learning algorithms for a diamond pricing model

Hybrid price prediction method combining TCN-BiGRU and attention mechanism for battery-grade lithium carbonate