J-stan-zh/project2-v15-direct-tree-model

Project2 V15 Direct-Tree Model

模型说明

• 模型版本：direct_tree_v15_e2_sitegroup_stronger
• 任务：基于历史长表数据，预测 E2/E3/E5 未来 48 小时
• 评估指标：SMAPE
• 本地验证文件：val_report.json

仓库文件

• direct_tree_bundle.pkl：训练好的模型 bundle
• val_report.json：当前版本验证结果
• 分析.md：完整版本迭代与实验说明

使用方式

1) 下载模型

from huggingface_hub import hf_hub_download

hf_hub_download(
    repo_id="J-stan-zh/project2-v15-direct-tree-model",
    filename="direct_tree_bundle.pkl",
    local_dir="models",
)

2) 推理

python run.py --history_data train_student_long.csv --model_dir models --output results/predictions.csv

3) 本地验证（可选）

python validate.py --history_data train_student_long.csv --model_dir models --report_file results/val_report_recheck.json

依赖

见 requirements.txt。