Comparison of Decision Tree Regression with Linear Regression Based on Prediction of Apple Stock Price

Abstract

Machine learning has been increasingly used in stock price prediction with outstanding success. Decision tree regression models and linear regression models are both important models for predicting stock prices. The paper use decision tree regression and linear regression models to predict the opening price, closing price, high price and low price of Apple's stock price data respectively. The prediction effects of the two models are evaluated by the indicators of goodness of fit, mean square error, root mean square error and mean absolute error, and the prediction effects of the two models are compared. This experimental concludes that the decision tree regression model has better and more advantageous prediction results compared to the linear regression model. This study has guiding significance for machine learning in predicting stock prices when choosing a basic model or a combination of models for prediction.