From 09d3fde34fc91195ed139c572903ba0b05130ce9 Mon Sep 17 00:00:00 2001
From: overlordmax <515704170@qq.com>
Date: Thu, 2 Jul 2020 14:18:15 +0800
Subject: [PATCH] add flen

---
 models/rank/flen/model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/rank/flen/model.py b/models/rank/flen/model.py
index 4557a031..5a9a26a1 100644
--- a/models/rank/flen/model.py
+++ b/models/rank/flen/model.py
@@ -107,7 +107,7 @@ class Model(ModelBase):
 
         return deep_input
 
-    def embeddingLayer(self, inputs):
+    def _embeddingLayer(self, inputs):
         emb_list = []
         in_len = len(inputs)
         for data in inputs:
@@ -138,7 +138,7 @@ class Model(ModelBase):
 
         field_wise_embeds_list = []
         for inputs in [self.user_inputs, self.item_inputs, self.contex_inputs]:
-            field_emb = self.embeddingLayer(inputs)
+            field_emb = self._embeddingLayer(inputs)
             field_wise_embeds_list.append(field_emb)
 
         dnn_input = fluid.layers.concat(
-- 
GitLab