Merge branch 'dev_david' of anweiguo/algorithm_platform into dev_awg

models_processing/model_koi/test.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+# @FileName  :test.py
+# @Time      :2025/3/5 18:11
+# @Author    :David
+# @Company: shenyang JY
+
+import tensorflow as tf
+print("TensorFlow 版本:", tf.__version__)
+print("GPU 是否可用:", tf.config.list_physical_devices('GPU'))

models_processing/model_koi/tf_bp.py

@@ -40,6 +40,7 @@ class BPHandler(object):
         model = Sequential([
             Dense(64, input_dim=opt.Model['input_size'], activation='relu'),  # 输入层和隐藏层，10个神经元
             Dense(32, activation='relu'),  # 隐藏层，8个神经元
+            Dense(16, activation='relu'),  # 隐藏层，8个神经元
             Dense(1, activation='linear')  # 输出层，1个神经元（用于回归任务）
         ])
         adam = optimizers.Adam(learning_rate=opt.Model['learning_rate'], beta_1=0.9, beta_2=0.999, epsilon=1e-7, amsgrad=True)

models_processing/model_koi/tf_bp_train.py

@@ -66,7 +66,6 @@ def model_training_bp():
         my_exception.replace("\n", "\t")
         result['msg'] = my_exception
     end_time = time.time()
-
     result['success'] = success
     result['args'] = args
     result['start_time'] = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start_time))

models_processing/model_koi/tf_cnn_train.py

@@ -63,12 +63,10 @@ def model_training_bp():
         insert_scaler_model_into_mongo(scaled_train_bytes, scaled_target_bytes, args)
         success = 1
     except Exception as e:
-        # my_exception = traceback.format_exc()
-        # my_exception.replace("\n", "\t")
-        # result['msg'] = my_exception
-        print("???", e)
+        my_exception = traceback.format_exc()
+        my_exception.replace("\n", "\t")
+        result['msg'] = my_exception
     end_time = time.time()
-
     result['success'] = success
     result['args'] = args
     result['start_time'] = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start_time))

models_processing/model_koi/tf_lstm_train.py

@@ -65,7 +65,6 @@ def model_training_bp():
         my_exception.replace("\n", "\t")
         result['msg'] = my_exception
     end_time = time.time()
-
     result['success'] = success
     result['args'] = args
     result['start_time'] = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start_time))