snapshot...

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)