Diamond occurs as one of the most precious and the strongest naturally occurring materials. It has always remained in demand because of its rarity, light refracting properties that make it appealing, and the urge to have it inlaid in the jewelry. Unlike silver and gold, finalizing its price is not an easy job. Its valuation is a significant aspect of the diamond industry, affected by various features like carat weight, clarity, cut, and color. This research paper studies the usage of regression techniques in machine learning for prediction of diamond prices. The dataset consists of features namely carat, color, cut and clarity, along with some physical dimensions like depth, x, y, z and table, which are either categorical or continuous. Some categorical features have been encoded into continuous (numerical) values so as to facilitate efficient and easier prediction. By analyzing the performance of different regression algorithms and assessing the outcomes based on the existing literature, the paper provides understanding of the effectiveness of regression analysis in price prediction of diamonds, clarifying the effect of different features on price variations. The regression algorithms implemented aim to obtain lower values of errors like MSE and RMSE, and greater value of R2 score. Random forest turned out to be the most efficient algorithm among them that effectively predicts diamond prices with an R2 score of 0.982 and minimum errors. Linear regression and decision tree also gave good R2 scores of 0.906 and 0.966, respectively.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Predicting Diamond Prices Using Regression in Machine Learning

  • Amarbirsingh Kuldipsingh Randhawa,
  • Ankush D. Sawarkar,
  • Sandeep B. Mundhe,
  • A. Waheed Deshmukh,
  • Pravin R. Sonarkar

摘要

Diamond occurs as one of the most precious and the strongest naturally occurring materials. It has always remained in demand because of its rarity, light refracting properties that make it appealing, and the urge to have it inlaid in the jewelry. Unlike silver and gold, finalizing its price is not an easy job. Its valuation is a significant aspect of the diamond industry, affected by various features like carat weight, clarity, cut, and color. This research paper studies the usage of regression techniques in machine learning for prediction of diamond prices. The dataset consists of features namely carat, color, cut and clarity, along with some physical dimensions like depth, x, y, z and table, which are either categorical or continuous. Some categorical features have been encoded into continuous (numerical) values so as to facilitate efficient and easier prediction. By analyzing the performance of different regression algorithms and assessing the outcomes based on the existing literature, the paper provides understanding of the effectiveness of regression analysis in price prediction of diamonds, clarifying the effect of different features on price variations. The regression algorithms implemented aim to obtain lower values of errors like MSE and RMSE, and greater value of R2 score. Random forest turned out to be the most efficient algorithm among them that effectively predicts diamond prices with an R2 score of 0.982 and minimum errors. Linear regression and decision tree also gave good R2 scores of 0.906 and 0.966, respectively.