metadata
language:
- en
metrics:
- mse
pipeline_tag: tabular-regression
library_name: keras
This model requires the following custom objects:
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) # output in range [0.5, 1.5]