notwa/optim
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

add onehot utility function

Browse source
This commit is contained in:
Connor Olding 2018-02-02 08:52:32 +01:00
parent e7783188bb
commit 39bbf27860
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onn/utility.py
View file

@ -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