David 3 hete
szülő
commit
b9e1ea5dd2
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      models_processing/model_tf/tf_multi_nwp_train.py

+ 1 - 1
models_processing/model_tf/tf_multi_nwp_train.py

@@ -33,7 +33,6 @@ def update_config():
     # features参数规则:1.有传入,解析,覆盖 2. 无传入,不覆盖,原始值
     # features参数规则:1.有传入,解析,覆盖 2. 无传入,不覆盖,原始值
     request_args['features'] = request_args['features'].split(',') if 'features' in request_args else current_config['features']
     request_args['features'] = request_args['features'].split(',') if 'features' in request_args else current_config['features']
     request_args['time_series'] = request_args.get('time_series', 1)
     request_args['time_series'] = request_args.get('time_series', 1)
-    request_args['zone'] = request_args['zone'].split(',')
     current_config = deep_update(current_config, request_args)
     current_config = deep_update(current_config, request_args)
 
 
     # 存储到请求上下文
     # 存储到请求上下文
@@ -55,6 +54,7 @@ def model_training_lstm():
     try:
     try:
         # ------------ 获取数据,预处理训练数据 ------------
         # ------------ 获取数据,预处理训练数据 ------------
         train_data = get_data_from_mongo(args)
         train_data = get_data_from_mongo(args)
+        train_data_1 = get_data_from_mongo(args)
         train_x, train_y, valid_x, valid_y, scaled_train_bytes, scaled_target_bytes, scaled_cap = dh.train_data_handler(train_data, time_series=args['time_series'])
         train_x, train_y, valid_x, valid_y, scaled_train_bytes, scaled_target_bytes, scaled_cap = dh.train_data_handler(train_data, time_series=args['time_series'])
         ts.opt.cap = round(scaled_cap, 2)
         ts.opt.cap = round(scaled_cap, 2)
         ts.opt.Model['input_size'] = len(dh.opt.features)
         ts.opt.Model['input_size'] = len(dh.opt.features)