sharktide commited on
Commit
3fa308b
·
verified ·
1 Parent(s): cacfa71

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ metrics:
5
+ - mse
6
+ pipeline_tag: tabular-classification
7
+ ---
8
+
9
+ This model requires the following custom objects:
10
+
11
+ ```python
12
+ import tensorflow as tf
13
+ from tensorflow.keras.saving import register_keras_serializable
14
+ from tensorflow.keras import layers, models, backend as K
15
+ import numpy as np
16
+
17
+ @register_keras_serializable()
18
+ def firetrust_activation(x):
19
+ return 0.5 + tf.sigmoid(x) # output in range [0.5, 1.5]
20
+ ```