Merge branch 'dev_david' of anweiguo/algorithm_platform into dev_awg

models_processing/model_tf/lstm.yaml

@@ -18,7 +18,7 @@ Model:
   shuffle_train_data: false
   test_data_fill: false
   time_step: 16
-  train_data_fill: false
+  train_data_fill: true
   use_cuda: false
   valid_data_rate: 0.15
 region: south

models_processing/model_tf/tf_bp_pre.py

@@ -101,7 +101,10 @@ def model_prediction_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10110)
+    serve(app, host="0.0.0.0", port=10110,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     print("server start!")
 
     # ------------------------测试代码------------------------

models_processing/model_tf/tf_bp_train.py

@@ -76,7 +76,13 @@ def model_training_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10111)
+    serve(
+        app,
+        host="0.0.0.0",
+        port=10111,
+        threads=8,  # 指定线程数（默认4，根据硬件调整）
+        channel_timeout=600  # 连接超时时间（秒）
+    )
     # 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', 'col_time': 'dateTime',

models_processing/model_tf/tf_cnn_pre.py

@@ -100,7 +100,10 @@ def model_prediction_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10112)
+    serve(app, host="0.0.0.0", port=10112,
+    threads = 8,  # 指定线程数（默认4，根据硬件调整）
+    channel_timeout = 600  # 连接超时时间（秒）
+          )
     print("server start!")
 
     # ------------------------测试代码------------------------

models_processing/model_tf/tf_cnn_train.py

@@ -80,7 +80,10 @@ def model_training_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10113)
+    serve(app, host="0.0.0.0", port=10113,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     # 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', 'col_time': 'dateTime',

models_processing/model_tf/tf_lstm_pre.py

@@ -99,7 +99,10 @@ def model_prediction_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10114)
+    serve(app, host="0.0.0.0", port=10114,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     print("server start!")
 
     # ------------------------测试代码------------------------

models_processing/model_tf/tf_lstm_train.py

@@ -76,7 +76,10 @@ def model_training_bp():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10115)
+    serve(app, host="0.0.0.0", port=10115,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     print("server start!")
     # args_dict = {"mongodb_database": 'realtimeDq', 'scaler_table': 'j00600_scaler', 'model_name': 'lstm1',
     # 'model_table': 'j00600_model', 'mongodb_read_table': 'j00600', 'col_time': 'dateTime',

models_processing/model_tf/tf_test_pre.py

@@ -16,7 +16,7 @@ import time, yaml
 model_lock = Lock()
 from itertools import chain
 from common.logs import Log
-from tf_lstm import TSHandler
+from tf_test import TSHandler
 # logger = Log('tf_bp').logger()
 logger = Log('tf_test').logger
 np.random.seed(42)  # NumPy随机种子
@@ -99,7 +99,10 @@ def model_prediction_test():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10116)
+    serve(app, host="0.0.0.0", port=10116,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     print("server start!")
 
     # ------------------------测试代码------------------------

models_processing/model_tf/tf_test_train.py

@@ -11,7 +11,7 @@ import traceback, uuid
 import logging, argparse
 from data_processing.data_operation.data_handler import DataHandler
 import time, yaml, threading
-from models_processing.model_tf.tf_lstm import TSHandler
+from models_processing.model_tf.tf_test import TSHandler
 from common.database_dml_koi import *
 from common.logs import Log
 logger = Log('tf_test').logger
@@ -76,7 +76,10 @@ def model_training_test():
 if __name__ == "__main__":
     print("Program starts execution!")
     from waitress import serve
-    serve(app, host="0.0.0.0", port=10117)
+    serve(app, host="0.0.0.0", port=10117,
+          threads=8,  # 指定线程数（默认4，根据硬件调整）
+          channel_timeout=600  # 连接超时时间（秒）
+          )
     print("server start!")
     # args_dict = {"mongodb_database": 'realtimeDq', 'scaler_table': 'j00600_scaler', 'model_name': 'lstm1',
     # 'model_table': 'j00600_model', 'mongodb_read_table': 'j00600', 'col_time': 'dateTime',