浏览代码

Update widedeep_v12_7: use w&d

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

+ 2 - 2
widedeep_v12_7.py

@@ -173,7 +173,7 @@ model_config {
   feature_groups: {
     group_name: 'deep'""")
 
-    for name in dense_features + sparse_features + tag_features + seq_features + nlp_features:
+    for name in top_dense_features + sparse_features + tag_features + seq_features + nlp_features:
         print(f"""    feature_names: '{name}'""")
 
     print("""    wide_deep: DEEP
@@ -263,5 +263,5 @@ export_config {
 """)
 
 
-deep_fm()
+wide_and_deep()
 config_export()