From 725996d8172bee93e5cdc938e4fc902e1996a03d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 3 May 2021 15:48:54 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 36a6c57f..97ee728d 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -381,8 +381,10 @@ class IcingIntensityNN:
fc = tf.keras.layers.BatchNormalization()(fc)
print(fc.shape)
- # activation = tf.nn.softmax # For multi-class
- activation = tf.nn.sigmoid # For binary
+ if NumClasses == 2:
+ activation = tf.nn.sigmoid # For binary
+ else:
+ activation = tf.nn.softmax # For multi-class
# Called logits, but these are actually probabilities see activation
logits = tf.keras.layers.Dense(NumLogits, activation=activation)(fc)
--
GitLab