@@ -290,7 +290,7 @@ def load_sick_data_from(sick_id, kind):
290290 line .append (texts [1 ].strip ())
291291 i .close ()
292292
293- j = open ('./tmp/sick.mapping_costs.txt' )
293+ j = open ('./tmp/sick.mapping_costs.new. txt' )
294294 scores = j .readlines ()
295295 for score in scores :
296296 if re .search ('^plain/sick_' + kind .lower ()+ '_' + sick_id + '.txt' , score ):
@@ -305,13 +305,13 @@ def load_sick_data_from(sick_id, kind):
305305def load_sick_data ():
306306 sick_train , sick_test = [], []
307307 for line in open ('./en/SICK.semeval.txt' ):
308- if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[11 ].strip () == 'TRAIN' :
308+ if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[- 1 ].strip () == 'TRAIN' :
309309 if load_sick_data_from (line .split ('\t ' )[0 ], 'TRAIN' ) is not None :
310310 sick_train .append (load_sick_data_from (line .split ('\t ' )[0 ], 'TRAIN' ))
311- if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[11 ].strip () == 'TRIAL' :
311+ if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[- 1 ].strip () == 'TRIAL' :
312312 if load_sick_data_from (line .split ('\t ' )[0 ], 'TRIAL' ) is not None :
313313 sick_train .append (load_sick_data_from (line .split ('\t ' )[0 ], 'TRIAL' ))
314- if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[11 ].strip () == 'TEST' :
314+ if line .split ('\t ' )[0 ] != 'pair_ID' and line .split ('\t ' )[- 1 ].strip () == 'TEST' :
315315 if load_sick_data_from (line .split ('\t ' )[0 ], 'TEST' ) is not None :
316316 sick_test .append (load_sick_data_from (line .split ('\t ' )[0 ], 'TEST' ))
317317 # if len(sick_train) == 10:
0 commit comments