jonathang commited on
Commit
06e7238
1 Parent(s): be2a07b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ class Sequence:
44
  max_length = 100
45
  padded_ie = pad_sequences([ie], maxlen=max_length, padding='post', truncating='post')
46
  all_ohe = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20] + [0]*(100-21))
47
- return to_categorical(np.array([padded_ie[0], all_ohe]))[0]
48
 
49
 
50
  def residual_block(data, filters, d_rate):
 
44
  max_length = 100
45
  padded_ie = pad_sequences([ie], maxlen=max_length, padding='post', truncating='post')
46
  all_ohe = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20] + [0]*(100-21))
47
+ return to_categorical(np.array([padded_ie[0], all_ohe]))[:1]
48
 
49
 
50
  def residual_block(data, filters, d_rate):