使用的是 models/base_model/params.json
,按照交叉验证,把输入的总数据集平分为4个数据组,train_batch
和test_batch
分别表示训练集和测试集使用的数据组序号,使用不同的训练集修改序号即可。
训练并预测输出
python src/main.py -model_dir models/base_model -mode train_pred
仅训练
python src/main.py -model_dir models/base_model -mode train
仅预测
python src/main.py -model_dir models/base_model -mode pred
按照params.json指定的训练集组序号,会在models/base_model/
下创建新的目录,由训练集组序号连接下划线命名,如0_1_2
,训练模型的ckpt文件会保存在这个目录,预测结果保存在其/predict
子目录