ПРОГНОЗУВАННЯ ВАРТОСТІ ТОВАРІВ ЗА ДОПОМОГОЮ РЕГРЕСІЙНОЇ МОДЕЛІ НА ОСНОВІ ФРЕЙМВОРКУ PYTORCH
DOI:
https://doi.org/10.17721/3041-2323.2024.297-303Ключові слова:
регресійні моделі, PyTorch, гіперпараметри, машинне навчання, MAE, MSEАнотація
Розроблено регресійну модель для прогнозування вартості товарів із використанням фреймворку PyTorch. Додатково побудовано дві класичні регресійні моделі машинного навчання – RandomForestRegressor і GradientBoostingRegressor. Реалізовано порівняння моделей. Визначено найкращу.
Посилання
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).
Завантаження
Опубліковано
Номер
Розділ
Ліцензія
Авторське право (c) 2024 Прикладні інформаційні системи та технології в цифровому суспільстві

Ця робота ліцензується відповідно до ліцензії Creative Commons Attribution 4.0 International License.