David преди 1 месец
родител
ревизия
d4d2304f76
променени са 2 файла, в които са добавени 5 реда и са изтрити 7 реда
  1. 3 6
      models_processing/model_koi/tf_bp_pre.py
  2. 2 1
      requirements.txt

+ 3 - 6
models_processing/model_koi/tf_bp_pre.py

@@ -17,7 +17,6 @@ model_lock = Lock()
 from itertools import chain
 from common.logs import Log
 from tf_bp import BPHandler
-# logger = Log('tf_bp').logger()
 logger = Log('tf_bp').logger
 np.random.seed(42)  # NumPy随机种子
 # tf.set_random_seed(42)  # TensorFlow随机种子
@@ -88,17 +87,15 @@ def model_prediction_bp():
 
 if __name__ == "__main__":
     print("Program starts execution!")
-    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-    logger = logging.getLogger("model_training_bp log")
     from waitress import serve
 
     serve(app, host="0.0.0.0", port=10110)
     print("server start!")
 
     # ------------------------测试代码------------------------
-    args_dict = {"mongodb_database": 'david_test', 'scaler_table': 'j00083_scaler', 'model_name': 'bp1.0.test',
-                 'model_table': 'j00083_model', 'mongodb_read_table': 'j00083_test', 'col_time': 'date_time', 'mongodb_write_table': 'j00083_rs',
-                 'features': 'speed10,direction10,speed30,direction30,speed50,direction50,speed70,direction70,speed90,direction90,speed110,direction110,speed150,direction150,speed170,direction170'}
+    # args_dict = {"mongodb_database": 'david_test', 'scaler_table': 'j00083_scaler', 'model_name': 'bp1.0.test',
+    #              'model_table': 'j00083_model', 'mongodb_read_table': 'j00083_test', 'col_time': 'date_time', 'mongodb_write_table': 'j00083_rs',
+    #              'features': 'speed10,direction10,speed30,direction30,speed50,direction50,speed70,direction70,speed90,direction90,speed110,direction110,speed150,direction150,speed170,direction170'}
     # args_dict['features'] = args_dict['features'].split(',')
     # arguments.update(args_dict)
     # dh = DataHandler(logger, arguments)

+ 2 - 1
requirements.txt

@@ -14,4 +14,5 @@ tensorflow==2.16.1
 matplotlib==3.10.0
 protobuf==3.20.3
 APScheduler==3.10.4
-paramiko==3.5.0
+paramiko==3.5.0
+PyYAML==6.0.1