浏览代码

Fix save model python script

Jérôme BUISINE 5 年之前
父节点
当前提交
10bfdf2af8
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      others/save_model_result_in_md_maxwell.py

+ 1 - 1
others/save_model_result_in_md_maxwell.py

@@ -53,7 +53,7 @@ def main():
 
     parser.add_argument('--interval', type=str, help='Interval value to keep from svd', default='"0, 200"')
     parser.add_argument('--model', type=str, help='.joblib or .json file (sklearn or keras model)')
-    parser.add_argument('--feature', type=str, help='Metric data choice', choices=cfg.feature_choices_labels)
+    parser.add_argument('--feature', type=str, help='Metric data choice', choices=cfg.features_choices_labels)
     parser.add_argument('--mode', type=str, help='Kind of normalization level wished', choices=cfg.normalization_choices)
 
     args = parser.parse_args()