|
@@ -13,9 +13,10 @@ from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau
|
|
import tensorflow as tf
|
|
import tensorflow as tf
|
|
from common.database_dml import get_data_from_mongo,insert_h5_model_into_mongo
|
|
from common.database_dml import get_data_from_mongo,insert_h5_model_into_mongo
|
|
from common.processing_data_common import missing_features,str_to_list
|
|
from common.processing_data_common import missing_features,str_to_list
|
|
-
|
|
|
|
|
|
+import threading
|
|
app = Flask('model_training_lightgbm——service')
|
|
app = Flask('model_training_lightgbm——service')
|
|
-
|
|
|
|
|
|
+# 创建一个锁对象
|
|
|
|
+lock = threading.Lock()
|
|
def rmse(y_true, y_pred):
|
|
def rmse(y_true, y_pred):
|
|
return tf.math.sqrt(tf.reduce_mean(tf.square(y_true - y_pred)))
|
|
return tf.math.sqrt(tf.reduce_mean(tf.square(y_true - y_pred)))
|
|
|
|
|
|
@@ -35,6 +36,7 @@ def create_sequences(data_features,data_target,time_steps):
|
|
|
|
|
|
|
|
|
|
def build_model(data, args):
|
|
def build_model(data, args):
|
|
|
|
+ tf.keras.backend.clear_session() # 清除当前的图和会话
|
|
col_time, time_steps,features,target = args['col_time'], int(args['time_steps']), str_to_list(args['features']),args['target']
|
|
col_time, time_steps,features,target = args['col_time'], int(args['time_steps']), str_to_list(args['features']),args['target']
|
|
if 'is_limit' in data.columns:
|
|
if 'is_limit' in data.columns:
|
|
data = data[data['is_limit']==False]
|
|
data = data[data['is_limit']==False]
|
|
@@ -120,5 +122,5 @@ if __name__=="__main__":
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
logger = logging.getLogger("model_training_lightgbm log")
|
|
logger = logging.getLogger("model_training_lightgbm log")
|
|
from waitress import serve
|
|
from waitress import serve
|
|
- serve(app, host="0.0.0.0", port=10096)
|
|
|
|
|
|
+ serve(app, host="0.0.0.0", port=10096,threads=4)
|
|
print("server start!")
|
|
print("server start!")
|