022008323

data_processing/data_operation/data_handler.py

@@ -233,5 +233,5 @@ class DataHandler(object):
             pre_x = np.array(pre_data)
         else:
             pre_x = self.get_predict_data([pre_data], features)
-            pre_x = np.array(pre_data)
+            pre_x = np.array(pre_x)
         return pre_x

models_processing/model_koi/tf_cnn_train.py

@@ -41,8 +41,10 @@ def model_training_bp():
     opt = argparse.Namespace(**args)
     logger.info(args_dict)
     try:
+        # ------------ 获取数据，预处理训练数据 ------------
         train_data = get_data_from_mongo(args_dict)
         train_x, valid_x, train_y, valid_y, scaled_train_bytes, scaled_target_bytes = dh.train_data_handler(train_data, opt)
+        # ------------ 训练模型，保存模型 ------------
         opt.Model['input_size'] = train_x.shape[2]
         bp_model = cnn.training(opt, [train_x, valid_x, train_y, valid_y])