Update 2_gen_dien_input.py

Author: 浅梦

code/2_gen_dien_input.py

@@ -136,7 +136,7 @@ def gen_sess_feature_dien(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]