Update 2_gen_dsin_input.py

Author: 浅梦

code/2_gen_dsin_input.py

@@ -138,7 +138,7 @@ def gen_sess_feature_dsin(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]
     sess_feature = ['cate_id', 'brand']