Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def onehot_encode(count, value):
- if count == 5:
- if value == 0:
- y = np.asarray([1,0,0,0,0])
- elif value == 1:
- y = np.asarray([0,1,0,0,0])
- elif value == 2:
- y = np.asarray([0,0,1,0,0])
- elif value == 3:
- y = np.asarray([0,0,0,1,0])
- elif value == 4:
- y = np.asarray([0,0,0,0,1])
- else:
- print "Error", s
- elif count == 9:
- if value == 0:
- y = np.asarray([1,0,0,0,0,0,0,0,0])
- elif value == 1:
- y = np.asarray([0,1,0,0,0,0,0,0,0])
- elif value == 2:
- y = np.asarray([0,0,1,0,0,0,0,0,0])
- elif value == 3:
- y = np.asarray([0,0,0,1,0,0,0,0,0])
- elif value == 4:
- y = np.asarray([0,0,0,0,1,0,0,0,0])
- elif value == 5:
- y = np.asarray([0,0,0,0,0,1,0,0,0])
- elif value == 6:
- y = np.asarray([0,0,0,0,0,0,1,0,0])
- elif value == 7:
- y = np.asarray([0,0,0,0,0,0,0,1,0])
- elif value == 8:
- y = np.asarray([0,0,0,0,0,0,0,0,1])
- else:
- print "Error", s
- elif count == 20:
- if value == 0:
- y = np.asarray([1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 1:
- y = np.asarray([0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 2:
- y = np.asarray([0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 3:
- y = np.asarray([0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 4:
- y = np.asarray([0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 5:
- y = np.asarray([0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 6:
- y = np.asarray([0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 7:
- y = np.asarray([0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 8:
- y = np.asarray([0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0])
- elif value == 9:
- y = np.asarray([0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0])
- elif value == 10:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0])
- elif value == 11:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0])
- elif value == 12:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0])
- elif value == 13:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0])
- elif value == 14:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0])
- elif value == 15:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0])
- elif value == 16:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0])
- elif value == 17:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0])
- elif value == 18:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0])
- elif value == 19:
- y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1])
- else:
- print "Error", value
- else:
- print "Not prepared for ", count, "of categories."
- y = "No"
- return
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement