From d4db83282d6499399772337154259c32bc84e815 Mon Sep 17 00:00:00 2001 From: RuyiLuo Date: Thu, 31 Mar 2022 11:06:20 +0800 Subject: [PATCH] Update features.py --- codes/funrec/features.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/codes/funrec/features.py b/codes/funrec/features.py index c2f25f35..4acdad42 100644 --- a/codes/funrec/features.py +++ b/codes/funrec/features.py @@ -220,8 +220,9 @@ class FeatureEncoder(object): for fc in self.feature_column_list: if isinstance(fc, DenseFeat): dense_feature_dict[fc.name] = \ - self.feature_input_layer_dict[fc] - + self.feature_input_layer_dict[fc.name] + + embedding_layers_dict = {} if len(self.sparse_feature_columns) > 0 or \ len(self.varlen_sparse_feature_columns) > 0: embedding_layers_dict = self.create_embedding_layers_dict(