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 }