项目重构版

运行顺序

1. python s1_deal_data.py
2. python s2_data_cut.py
3. python s3_feature_get.py
4. python s4_baseline.py
5. python s4_train_Linear.py
6. python s4_train_RF.py
7. python s4_train_MLP.py
8. python s4_train_RNN.py
9. python s4_train_LSTM.py
10. python s4_train_Emb_LSTM.py
11. python s5_get_indices.py
12. python s6_ablation_study.py

目录说明

• deal_data_feature/attr：属性变量版本特征
• deal_data_feature/onehot：one-hot 消融版本特征
• results：s4 的验证集预测结果
• args：最优参数
• models：模型文件
• indices：s5 生成的指标

冷启动验证集

默认会优先尝试以下地区*作物：

• 高州 * Lychee
• 徐闻 * Pineapple
• 化州 * Longan
• 丰顺 * Rice
• 大埔 * Tea

如果某些组合在特征过滤后不存在，会自动从候选列表补齐。