Commit fd7b1b49 authored by 郭羽's avatar 郭羽

模型删除 second 特征

parent e5b1c88a
...@@ -194,7 +194,7 @@ ...@@ -194,7 +194,7 @@
# #677 # #677
# #
# #659 # #659
# #
# def tett(): # def tett():
# # keys = ['rims:tag_v3:coldstart:'] #177 # # keys = ['rims:tag_v3:coldstart:'] #177
# # keys = ['doris:tag_v3:coldstart:'] #471 # # keys = ['doris:tag_v3:coldstart:'] #471
......
...@@ -78,7 +78,7 @@ def getTrainColumns(train_columns,data_vocab): ...@@ -78,7 +78,7 @@ def getTrainColumns(train_columns,data_vocab):
# 离散特征 # 离散特征
for feature in train_columns: for feature in train_columns:
if data_vocab.get(feature): 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 continue
cat_col = tf.feature_column.categorical_column_with_vocabulary_list(key=feature,vocabulary_list=data_vocab[feature]) 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