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