diff options
| author | ZJaume <jzaragoza@prompsit.com> | 2022-03-01 19:23:30 +0300 |
|---|---|---|
| committer | ZJaume <jzaragoza@prompsit.com> | 2022-03-01 19:25:59 +0300 |
| commit | cdc27e587f8c90ec908e1610047ac561ece6808d (patch) | |
| tree | 7def878c96f2b2a73180ce561211a773492a5d85 | |
| parent | 28bffaa3799ab9dca97a5ba5a4398f00b07c5d37 (diff) | |
Explicit input shape when calibrating with TF 2.8
| -rw-r--r-- | bicleaner_ai/models.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index 4712864..902a648 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -73,6 +73,7 @@ def calibrate_output(y_true, y_pred): # dense with single neuron and bias A*x + B with tf.device("/cpu:0"): model = tf.keras.Sequential([ + tf.keras.layers.Input(shape=(1,)), tf.keras.layers.Dense(1, activation='sigmoid'), ]) loss = BinaryCrossentropy(reduction=tf.keras.losses.Reduction.SUM) |