Explorar el Código

修改nlp特征配置

xueyiming hace 3 semanas
padre
commit
41ad30b719
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  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():