Update 2_gen_din_input.py

Author: 浅梦

code/2_gen_din_input.py

@@ -86,7 +86,7 @@ def gen_sess_feature_din(row):
     mms = StandardScaler()
     data[dense_features] = mms.fit_transform(data[dense_features])
 
-    sparse_feature_list = [SingleFeat(feat, data[feat].nunique(
+    sparse_feature_list = [SingleFeat(feat, data[feat].max(
     ) + 1) for feat in sparse_features + ['cate_id', 'brand']]
 
     dense_feature_list = [SingleFeat(feat, 1) for feat in dense_features]