Create custom_objects.py

custom_objects.py

@@ -0,0 +1,13 @@
+import tensorflow as tf
+from tensorflow.keras.saving import register_keras_serializable
+from tensorflow.keras import layers, models, backend as K
+import numpy as np
+
+
+def firetrust_activation(x):
+    return 0.5 + tf.sigmoid(x)
+
+CUSTOM_OBJECTS = {
+    "firetrust_activation": firetrust_activation,
+    "mse": tf.keras.losses.MeanSquaredError,
+}