https://github.com/william-zhan-bot/sport_shoes_predict_by_ai

以ai模型對球鞋售價進行預測 Sneakers Price Predict Based on ML model
https://github.com/william-zhan-bot/sport_shoes_predict_by_ai

ai decision-tree-regression decision-trees machine-learning python sklearn

Last synced: 3 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/william-zhan-bot/sport_shoes_predict_by_ai
• Owner: William-Zhan-bot
• Created: 2024-09-09T03:06:26.000Z (19 days ago)
• Default Branch: main
• Last Pushed: 2024-09-09T03:18:00.000Z (19 days ago)
• Last Synced: 2024-09-25T00:09:39.981Z (3 days ago)
• Topics: ai, decision-tree-regression, decision-trees, machine-learning, python, sklearn
• Language: Jupyter Notebook
• Homepage:
• Size: 1.69 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# sport_shoes_predict

以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必須先跑過預測產生模型才可順利執行