rm call to private Model._make_predict_function

Discussions on why it isn't recommended to call: * keras-team/keras#6124 * keras-team/keras#13116
mihajenko · May 5, 2020 · fae6a90 · fae6a90
1 parent 4121bd4
commit fae6a90
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ivis/ivis.py b/ivis/ivis.py
@@ -398,11 +398,11 @@ def load_model(self, folder_path):
         self.__dict__ = ivis_config
 
         loss_function = triplet_loss(self.distance, self.margin)
-        self.model_ = load_model(os.path.join(folder_path, 'ivis_model.h5'),
-                                 custom_objects={'tf': tf,
-                                                 loss_function.__name__: loss_function })
+        self.model_: Model = load_model(
+            os.path.join(folder_path, 'ivis_model.h5'),
+            custom_objects={'tf': tf, loss_function.__name__: loss_function}
+        )
         self.encoder = self.model_.layers[3]
-        self.encoder._make_predict_function()
 
         # If a supervised model exists, load it
         supervised_path = os.path.join(folder_path, 'supervised_model.h5')