Advertisement
Guest User

Untitled

a guest
Mar 29th, 2017
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.65 KB | None | 0 0
  1. def onehot_encode(count, value):
  2.     if count == 5:
  3.         if value == 0:
  4.             y = np.asarray([1,0,0,0,0])
  5.         elif value == 1:
  6.             y = np.asarray([0,1,0,0,0])
  7.         elif value == 2:
  8.             y = np.asarray([0,0,1,0,0])
  9.         elif value == 3:
  10.             y = np.asarray([0,0,0,1,0])
  11.         elif value == 4:
  12.             y = np.asarray([0,0,0,0,1])
  13.         else:
  14.             print "Error", s
  15.     elif count == 9:
  16.         if value == 0:
  17.             y = np.asarray([1,0,0,0,0,0,0,0,0])
  18.         elif value == 1:
  19.             y = np.asarray([0,1,0,0,0,0,0,0,0])
  20.         elif value == 2:
  21.             y = np.asarray([0,0,1,0,0,0,0,0,0])
  22.         elif value == 3:
  23.             y = np.asarray([0,0,0,1,0,0,0,0,0])
  24.         elif value == 4:
  25.             y = np.asarray([0,0,0,0,1,0,0,0,0])
  26.         elif value == 5:
  27.             y = np.asarray([0,0,0,0,0,1,0,0,0])
  28.         elif value == 6:
  29.             y = np.asarray([0,0,0,0,0,0,1,0,0])
  30.         elif value == 7:
  31.             y = np.asarray([0,0,0,0,0,0,0,1,0])
  32.         elif value == 8:
  33.             y = np.asarray([0,0,0,0,0,0,0,0,1])
  34.         else:
  35.             print "Error", s
  36.     elif count == 20:
  37.         if value == 0:
  38.             y = np.asarray([1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  39.         elif value == 1:
  40.             y = np.asarray([0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  41.         elif value == 2:
  42.             y = np.asarray([0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  43.         elif value == 3:
  44.             y = np.asarray([0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  45.         elif value == 4:
  46.             y = np.asarray([0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  47.         elif value == 5:
  48.             y = np.asarray([0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
  49.         elif value == 6:
  50.             y = np.asarray([0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0,0])  
  51.         elif value == 7:
  52.             y = np.asarray([0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0,0])  
  53.         elif value == 8:
  54.             y = np.asarray([0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0,0])
  55.         elif value == 9:
  56.             y = np.asarray([0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0,0])
  57.         elif value == 10:
  58.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0,0])
  59.         elif value == 11:
  60.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0,0])
  61.         elif value == 12:
  62.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,0])
  63.         elif value == 13:
  64.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0])
  65.         elif value == 14:
  66.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0,0])
  67.         elif value == 15:
  68.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0,0])
  69.         elif value == 16:
  70.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0,0])
  71.         elif value == 17:
  72.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0,0])
  73.         elif value == 18:
  74.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,0])
  75.         elif value == 19:
  76.             y = np.asarray([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1])
  77.         else:
  78.             print "Error", value
  79.     else:
  80.         print "Not prepared for ", count, "of categories."
  81.         y = "No"       
  82.     return
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement