diff --git a/README.md b/README.md index ec44e80..5ac8773 100644 --- a/README.md +++ b/README.md @@ -3,9 +3,12 @@ ### 使用说明 1. 下载: 使用`git clone https://bdgit.educoder.net/pseuc2htk/NeuralCF.git`命令下载,或者直接下载zip包解压使用。 + 2. 环境配置: 要求python>=3.6,并配置好GPU环境,其他环境配置如下: > pytorch>=1.4 +cuda=10.0 +cuDNN =7.6 sklearn numpy pandas @@ -18,4 +21,7 @@ pandas `python train_test.py` 进行训练和推理。 +5. 运行`train_test.py`中的推理部分,对给出的测试数据进行预测并打印出预测值。 + +6. 将打印出的预测值粘贴到头歌测试平台该关卡下的`result/result.txt`中进行评测;