训练数据与测试数据都要放在Data文件夹下,由于训练数据较大,我暂未上传。
在进行参数选择的训练时,请使用train_para.py文件进行训练,超参选择范围可以在config.py文件中进行修改,在终端输入python train_para.py即可进行训练。
在进行单次训练时,可以使用train.py文件进行训练,超参可以在train.py文件中进行修改,在终端输入python train.py即可进行训练。
最优模型保存为best_model.pkl,与测试与训练代码在同一路径下,进行测试时,在终端输入test_model.py即可进行测试。