FORECASTING THE COST OF GOODS USING A REGRESSION MODEL BASED ON THE PYTORCH FRAMEWORK
DOI:
https://doi.org/10.17721/3041-2323.2024.297-303Keywords:
regression models, PyTorch, hyperparameters, machine learning, MAE, MSEAbstract
In this work, a regression model for predicting the cost of goods using the PyTorch framework has been developed. Additionally, two classical machine learning regression models – RandomForestRegressor and GradientBoostingRegressor – were constructed. A comparison of the models was carried out, and the best one was identified.
References
GradientBoostingRegressor. (n. d.). Scikit-learn documentation. https://scikitlearn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingRegressor.html (Accessed: 16.09.2024).
Laptop Prices. (n. d.). Kaggle. https://www.kaggle.com/datasets/owm4096/laptopprices (Accessed: 16.09.2024).
Mean Absolute Error. (n. d.). Scikit-learn documentation. https://scikitlearn.org/stable/modules/generated/sklearn.metrics.mean_absolute_error.html (Accessed: 16.09.2024).
Mean Squared Error. (n. d.). Scikit-learn documentation. https://scikitlearn.org/stable/modules/generated/sklearn.metrics.mean_squared_error.html (Accessed: 16.09.2024).
PyTorch. (n. d.). Official website. https://pytorch.org/ (Accessed: 16.09.2024).
R² Score. (n. d.). Scikit-learn documentation. https://scikit-learn.org/stable/modules/-generated/sklearn.metrics.r2_score.html (Accessed: 16.09.2024).
RandomForestRegressor. (n. d.). Scikit-learn documentation. https://scikitlearn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html (Accessed: 16.09.2024).
Randomized Search CV. (n. d.). Scikit-learn documentation. https://scikitlearn.org/stable/modules/generated/sklearn.model_selection.RandomizedSearchCV.html (Accessed: 17.09.2024).
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Applied information systems and technologies in the digital society
This work is licensed under a Creative Commons Attribution 4.0 International License.
