giving class label from probabilities

CNNmodel/myutil.py

+#!/usr/bin/env python
+'''
+Author : Gunasinghe M.D.
+IT NUM : IT17043342
+
+'''
+
+
+# Giving class label from probabilities
+from keras.utils import np_utils
+import numpy as np
+
+
+def probas_to_classes(y_pred):
+    if len(y_pred.shape) > 1 and y_pred.shape[1] > 1:
+        return categorical_probas_to_classes(y_pred)
+    return np.array([1 if p > 0.5 else 0 for p in y_pred])
+
+
+def categorical_probas_to_classes(p):
+    return np.argmax(p, axis=1)