David 3 maanden geleden
bovenliggende
commit
9a65c67c16
2 gewijzigde bestanden met toevoegingen van 2 en 2 verwijderingen
  1. 1 1
      models_processing/model_tf/tf_test_pre.py
  2. 1 1
      models_processing/model_tf/tf_test_train.py

+ 1 - 1
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随机种子

+ 1 - 1
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