Merge pull request #55 from NACLab/dev

fixed adj-acc bug in analyze_scores in metrics

Author: ago109

ngclearn/utils/metric_utils.py

@@ -122,7 +122,7 @@ def analyze_scores(mu, y, extract_label_indx=True): ## examines classifcation st
             confusion matrix, precision, recall, misses (empty predictions/all-zero rows),
             accuracy, adjusted-accuracy (counts all misses as incorrect)
         """
-    miss_mask = (jnp.sum(mu, axis=1, keepdims=True) == 0.) * 1.
+    miss_mask = (jnp.sum(mu, axis=1) == 0.) * 1.
     misses = jnp.sum(miss_mask) ## how many misses?
     labels = y
     if extract_label_indx:
@@ -133,7 +133,7 @@ def analyze_scores(mu, y, extract_label_indx=True): ## examines classifcation st
     recall = recall_score(labels, guesses, average='macro')
     ## produce accuracy score measurements
     guess = jnp.argmax(mu, axis=1) ## gather all model/output guesses
-    equality_mask = jnp.equal(guess, labels)
+    equality_mask = jnp.equal(guess, labels) * 1.
     ### compute raw accuracy
     acc = jnp.sum(equality_mask) / (y.shape[0] * 1.)
     ### compute hit-masked accuracy (adjusted accuracy