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
+
+@register_keras_serializable()
+def floodtrust_activation(x):
+    return 0.5 + tf.sigmoid(x)
+
+CUSTOM_OBJECTS = {
+    "floodtrust_activation": floodtrust_activation,
+    "mse": tf.keras.losses.MeanSquaredError
+}