Research on the Diamond Price Prediction based on Linear Regression, Decision Tree and Random Forest

Abstract

Diamonds are the symbol of the pure and indestructible love and the luxury that people have always sought after. However, because people know less about diamonds, they often only rely on the introduction of salespeople and jewelers in diamond trading. Therefore, it is difficult for consumers to buy diamonds of equal value and price. To solve this problem, this paper uses Multiple Linear Regression model, Decision Tree Regression model and Random Forest Regression model to predict diamond prices based on various diamond evaluation metrics in data set, so that consumers can intuitively learn about the normal price of the evaluation metrics of selected diamonds. Through this paper, it is found that the Random Forest Regression model has the best fitting and predictive ability in diamond prediction task, which is also the most recommended model.