04071505

app/model/main.py

@@ -111,7 +111,7 @@ def main():
     # 添加参数
     parser.add_argument("input_file", help="输入文件路径")    # 第一个位置参数
 
-    parser.add_argument("--model_name", default="cnn", help='选择短期模型')    # 第二个位置参数
+    parser.add_argument("--model_name", default="lstm", help='选择短期模型')    # 第二个位置参数
     # 解析参数
     args = parser.parse_args()