diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 95876f073e146ba7d43f245972d354380f3ae914..e9d619c8e470c382651ade376068325119b251d4 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -359,7 +359,12 @@ class IcingIntensityFCN:
         nda[np.logical_and(nda >= 6000, nda < 8000)] = 3
         nda[np.logical_and(nda >= 8000, nda < 15000)] = 4
 
+        # nda[np.logical_and(nda >= 0, nda < 2000)] = 0
+        # nda[np.logical_and(nda >= 2000, nda < 8000)] = 1
+        # nda[np.logical_and(nda >= 8000, nda < 15000)] = 2
+
         nda = tf.one_hot(nda, 5).numpy()
+        # nda = tf.one_hot(nda, 3).numpy()
         nda = np.expand_dims(nda, axis=1)
         nda = np.expand_dims(nda, axis=1)