correcting x -- thanks to huy

ganeshjawahar · Aug 10, 2017 · 5b77702 · 5b77702
1 parent 584b4ea
commit 5b77702
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -171,7 +171,7 @@ def train(self, data):
 
  for b in xrange(self.batch_size):
  m = rand_idx[cur]
- x[b][0] = target_data[0]
+ x[b][0] = target_data[m]
  target[b][target_label[m]] = 1
  time[b,:len(source_loc_data[m])] = source_loc_data[m]
  context[b,:len(source_data[m])] = source_data[m]