소스 검색

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()