add onehot utility function

2018-02-02 08:52:32 +01:00 · 2018-02-02 08:52:32 +01:00 · 39bbf27860
parent e7783188bb
commit 39bbf27860
1 changed files with 8 additions and 0 deletions
--- a/onn/utility.py
+++ b/onn/utility.py
@ -28,6 +28,14 @@ def lower_priority():
        os.nice(1)


+def onehot(y):
+    unique = np.unique(y)
+    Y = np.zeros((y.shape[0], len(unique)), dtype=np.int8)
+    offsets = np.arange(len(y)) * len(unique)
+    Y.flat[offsets + y.flat] = 1
+    return Y
+
+
 # more

 _log_was_update = False