ПРОГНОЗУВАННЯ ВАРТОСТІ ТОВАРІВ ЗА ДОПОМОГОЮ РЕГРЕСІЙНОЇ МОДЕЛІ НА ОСНОВІ ФРЕЙМВОРКУ PYTORCH

Автор(и)

  • Олег ПУРСЬКИЙ Державний торговельно-економічний університет image/svg+xml Автор
  • Тетяна ФІЛІМОНОВА Державний торговельно-економічний університет image/svg+xml Автор
  • Анна СЕЛІВАНОВА Державний торговельно-економічний університет image/svg+xml Автор
  • Андрій НЕЧЕПУРЕНКО Державний торговельно-економічний університет image/svg+xml Автор
  • Тетяна ДУБОВИК Державний торговельно-економічний університет image/svg+xml Автор

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).

Title

Завантаження

Опубліковано

01.10.2024