File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 1515runner :
1616 train_data_dir : " ../../../datasets/movielens_pinterest_NCF/big_train"
1717 train_reader_path : " movielens_reader" # importlib format
18- train_batch_size : 1024
18+ train_batch_size : 256
1919 model_save_path : " output_model_ncf"
2020
2121 use_gpu : False
Original file line number Diff line number Diff line change @@ -81,7 +81,7 @@ python -u ../../../tools/static_infer.py -m config.yaml
8181
8282| 模型 | HR@10 | NDCG@10 | batch_size | epoch_num| Time of each epoch |
8383| :------| :------ | :------ | :------ | :------| :------ |
84- | NCF_NeuMF | 0.58 | 0.33 | 1024 | 20 | 约20分钟 |
84+ | NCF_NeuMF | 0.58 | 0.33 | 256 | 20 | 约20分钟 |
8585
86861 . 确认您当前所在目录为PaddleRec/models/recall/ncf
87872 . 进入paddlerec/datasets/movielens_pinterest_NCF目录下,执行该脚本,会从国内源的服务器上下载我们预处理完成的movielens和pinterest全量数据集,并解压到指定文件夹。
You can’t perform that action at this time.
0 commit comments