From c40a0c65f1304527f5a22429859d2cc9f61cf9b5 Mon Sep 17 00:00:00 2001
From: frankwhzhang <frankwhzhang@126.com>
Date: Mon, 1 Jun 2020 16:32:22 +0800
Subject: [PATCH] fix bugs

---
 models/multitask/readme.md    | 1 -
 models/recall/ncf/config.yaml | 2 +-
 models/recall/ncf/model.py    | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/models/multitask/readme.md b/models/multitask/readme.md
index 76554172..07a6c01d 100755
--- a/models/multitask/readme.md
+++ b/models/multitask/readme.md
@@ -69,7 +69,6 @@ sh run.sh
 ### 训练
 ```
 cd modles/multitask/mmoe # 进入选定好的排序模型的目录 以MMoE为例
-python -m paddlerec.run -m paddlerec.models.multitask.mmoe # 使用内置配置
 python -m paddlerec.run -m ./config.yaml # 自定义修改超参后，指定配置文件，使用自定义配置
 ```
 
diff --git a/models/recall/ncf/config.yaml b/models/recall/ncf/config.yaml
index 608c712d..16d298b1 100644
--- a/models/recall/ncf/config.yaml
+++ b/models/recall/ncf/config.yaml
@@ -30,7 +30,7 @@ hyper_parameters:
   num_users: 6040
   num_items: 3706
   latent_dim: 8
-  layers: [64, 32, 16, 8]
+  fc_layers: [64, 32, 16, 8]
   optimizer: 
     class: adam
     learning_rate: 0.001
diff --git a/models/recall/ncf/model.py b/models/recall/ncf/model.py
index 78d23c69..bc8b71cd 100644
--- a/models/recall/ncf/model.py
+++ b/models/recall/ncf/model.py
@@ -28,7 +28,7 @@ class Model(ModelBase):
         self.num_users = envs.get_global_env("hyper_parameters.num_users")
         self.num_items = envs.get_global_env("hyper_parameters.num_items")
         self.latent_dim = envs.get_global_env("hyper_parameters.latent_dim")
-        self.layers = envs.get_global_env("hyper_parameters.layers")
+        self.layers = envs.get_global_env("hyper_parameters.fc_layers")
 
     def input_data(self, is_infer=False, **kwargs):
         user_input = fluid.data(
-- 
GitLab