custom_objects.py

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 firetrust_activation(x):
    return 0.5 + tf.sigmoid(x)

CUSTOM_OBJECTS = {
    "firetrust_activation": firetrust_activation,
    "mse": tf.keras.losses.MeanSquaredError
}