Commit fd7b1b49 authored by 郭羽's avatar 郭羽

模型删除 second 特征

parent e5b1c88a
......@@ -78,7 +78,7 @@ def getTrainColumns(train_columns,data_vocab):
# 离散特征
for feature in train_columns:
if data_vocab.get(feature):
if feature.startswith("user__second") and not feature.endswith("1"):
if feature.startswith("user__second") or feature.startswith("item_second"):
continue
cat_col = tf.feature_column.categorical_column_with_vocabulary_list(key=feature,vocabulary_list=data_vocab[feature])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment