Update simple_cerebros_random_search.py

david-thrower · Jun 12, 2024 · 2b6ccc4 · 2b6ccc4
1 parent 36b6663
commit 2b6ccc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
@@ -573,7 +573,7 @@ def run_random_search(self):
     def get_best_model(self):
         best_model_path = self.best_model_path
         best_model =\
-            tensorflow.keras.models.clone_model(
+            tf.keras.models.clone_model(
                 tf.keras.models.load_model(best_model_path))
         # best_model = tf.keras.models.load_model(self.best_model_path)
         return best_model