@ -6,6 +6,8 @@
2. 环境配置:
要求python>=3.6,并配置好GPU环境,其他环境配置如下:
> pytorch>=1.4
cuda=10.0
cuDNN =7.6
sklearn
numpy
pandas
@ -17,4 +19,5 @@ pandas
在`./tool`文件夹下,配置`./tool/train_test.py`下训练参数后,运行:
`python train_test.py`
进行训练和推理。
5. 运行`train_test.py`中的推理部分,对给出的测试数据进行预测并打印出预测值。
6. 将打印出的预测值粘贴到测试平台的`result/result.txt`中进行评测;