diff --git a/models/rank/flen/model.py b/models/rank/flen/model.py index 4557a031c9983f92dee6ceb4806a57e5c404e4b0..5a9a26a1386899d094fdca9ce72fc59709e1dcee 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(