浏览代码

修改nlp特征配置

xueyiming 3 周之前
父节点
当前提交
41ad30b719
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      widedeep_v12_7.py

+ 2 - 2
widedeep_v12_7.py

@@ -152,10 +152,10 @@ for name in tag_features + seq_features:
 for name in nlp_features:
     print(f"""feature_configs {{
   input_names: "{name}"
-  feature_type: TagFeature
+  feature_type: RawFeature
   hash_bucket_size: 1000000
-  embedding_dim: 6
   separator: '|'
+  raw_input_dim: 200
 }}""")
 
 def wide_and_deep():