Skip to content

以ai模型對球鞋售價進行預測 Sneakers Price Predict Based on ML model

Notifications You must be signed in to change notification settings

William-Zhan-bot/sport_shoes_predict_by_AI

Repository files navigation

sport_shoes_predict

website: https://sportshoes-predict-website.onrender.com/

以Kaggle 2020~2023 之球鞋基本資料,訓練決策樹回歸模型進行球鞋售價預測 Kaggle: https://www.kaggle.com/datasets/rkiattisak/shoe-prices-dataset

資料庫欄位: 品牌 款式名稱 出產年份 主打族群 尺寸 功能類別 材質 主要顏色 售價 數量: 100筆

選用 Feature: 品牌 主打族群 尺寸 功能類別 材質 主要顏色 Label: 售價(USD)

算法選用: 以線性回歸成效過低,r平方僅為0.15 改以決策樹回歸進行評估,r平方達到0.7

資料清理: 皆為類別資料,先對所有feature進行one-hot encoding後開始以決策樹回歸進行訓練

模型輸出後檔案為: decision_tree_model.pkl 啟動predict.ipynb必須先跑過預測產生模型才可順利執行

About

以ai模型對球鞋售價進行預測 Sneakers Price Predict Based on ML model

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published