Merge pull request #7 from MSD-IRIMAS/enh-keras-saved-models

hadifawaz1999 · web-flow · commit 183f6c331941 · 2025-02-14T12:47:14.000+01:00
[ENH] Replace .hdf5 saved model with .keras for new version compatibility
diff --git a/classifiers/lite.py b/classifiers/lite.py
@@ -265,7 +265,7 @@ def build_model(self):
             monitor="loss", factor=0.5, patience=50, min_lr=1e-4
         )
         model_checkpoint = tf.keras.callbacks.ModelCheckpoint(
-            filepath=self.output_directory + "best_model.hdf5",
+            filepath=self.output_directory + "best_model.keras",
             monitor="loss",
             save_best_only=True,
         )
@@ -446,7 +446,7 @@ def _fit(xtrain, ytrain, xval, yval, plot_test):
     def predict(self, xtest, ytest):
 
         model = tf.keras.models.load_model(
-            self.output_directory + "best_model.hdf5", compile=False
+            self.output_directory + "best_model.keras", compile=False
         )
 
         start_time = time.time()
diff --git a/classifiers/litemv.py b/classifiers/litemv.py
@@ -270,7 +270,7 @@ def build_model(self):
             monitor="loss", factor=0.5, patience=50, min_lr=1e-4
         )
         model_checkpoint = tf.keras.callbacks.ModelCheckpoint(
-            filepath=self.output_directory + "best_model.hdf5",
+            filepath=self.output_directory + "best_model.keras",
             monitor="loss",
             save_best_only=True,
         )
@@ -451,7 +451,7 @@ def _fit(xtrain, ytrain, xval, yval, plot_test):
     def predict(self, xtest, ytest):
 
         model = tf.keras.models.load_model(
-            self.output_directory + "best_model.hdf5", compile=False
+            self.output_directory + "best_model.keras", compile=False
         )
 
         start_time = time.time()

Original file line number	Diff line number	Diff line change
`@@ -265,7 +265,7 @@ def build_model(self):`
`265`	`265`	`monitor="loss", factor=0.5, patience=50, min_lr=1e-4`
`266`	`266`	`)`
`267`	`267`	`model_checkpoint = tf.keras.callbacks.ModelCheckpoint(`
`268`		`- filepath=self.output_directory + "best_model.hdf5",`
	`268`	`+ filepath=self.output_directory + "best_model.keras",`
`269`	`269`	`monitor="loss",`
`270`	`270`	`save_best_only=True,`
`271`	`271`	`)`
`@@ -446,7 +446,7 @@ def _fit(xtrain, ytrain, xval, yval, plot_test):`
`446`	`446`	`def predict(self, xtest, ytest):`
`447`	`447`
`448`	`448`	`model = tf.keras.models.load_model(`
`449`		`- self.output_directory + "best_model.hdf5", compile=False`
	`449`	`+ self.output_directory + "best_model.keras", compile=False`
`450`	`450`	`)`
`451`	`451`
`452`	`452`	`start_time = time.time()`
Original file line number	Diff line number	Diff line change
`@@ -270,7 +270,7 @@ def build_model(self):`
`270`	`270`	`monitor="loss", factor=0.5, patience=50, min_lr=1e-4`
`271`	`271`	`)`
`272`	`272`	`model_checkpoint = tf.keras.callbacks.ModelCheckpoint(`
`273`		`- filepath=self.output_directory + "best_model.hdf5",`
	`273`	`+ filepath=self.output_directory + "best_model.keras",`
`274`	`274`	`monitor="loss",`
`275`	`275`	`save_best_only=True,`
`276`	`276`	`)`
`@@ -451,7 +451,7 @@ def _fit(xtrain, ytrain, xval, yval, plot_test):`
`451`	`451`	`def predict(self, xtest, ytest):`
`452`	`452`
`453`	`453`	`model = tf.keras.models.load_model(`
`454`		`- self.output_directory + "best_model.hdf5", compile=False`
	`454`	`+ self.output_directory + "best_model.keras", compile=False`
`455`	`455`	`)`
`456`	`456`
`457`	`457`	`start_time = time.time()`