Add flat pred function

HessTaha · May 31, 2020 · c0ca7a3 · c0ca7a3
1 parent 4f5770c
commit c0ca7a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/utils.py b/src/utils.py
@@ -21,6 +21,8 @@
 from sklearn.model_selection import train_test_split
 from keras.utils import to_categorical
 
+def flat_pred(preds, threshold):
+    return np.where(preds>threshold, 1, 0)
 
 def set_seed(seed):
     random.seed(seed)