Update simple_cerebros_random_search.py

Author: david-thrower

cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py

@@ -572,10 +572,14 @@ def run_random_search(self):
 
     def get_best_model(self):
         best_model_path = self.best_model_path
+        # best_model =\
+        #     tf.keras.models.clone_model(
+        #         tf.keras.models.load_model(best_model_path, custom_objects={"TernaryDenseLayer": units.units.TernaryDenseLayer}))
         best_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)
+            tf.keras.models.load_model(
+                self.best_model_path,
+                custom_objects=\
+                    {"TernaryDenseLayer": units.units.TernaryDenseLayer})
         return best_model
 
 # ->