bug fixes in recasting
@@ -1217,12 +1217,13 @@ class RecastNLITask(PairClassificationTask):
self.val_data_text[0] + self.val_data_text[1]
def load_data(self, path, max_seq_len):
+ log.info("In load_data")
tr_data = load_tsv(os.path.join(path, 'train.tsv'), max_seq_len,
- s1_idx=1, s2_idx=2, skip_rows=0)
+ s1_idx=1, s2_idx=2, skip_rows=0, targ_idx=3)
val_data = load_tsv(os.path.join(path, 'dev.tsv'), max_seq_len,
- s1_idx=0, s2_idx=1, skip_rows=0)
+ s1_idx=0, s2_idx=1, skip_rows=0, tar_idx=3)
te_data = load_tsv(os.path.join(path, 'test.tsv'), max_seq_len,
- s1_idx=1, s2_idx=2, skip_rows=0)
+ s1_idx=1, s2_idx=2, skip_rows=0, targ_idx=3)
self.train_data_text = tr_data
self.val_data_text = val_data
