TestTaker commited on
Commit
b393949
1 Parent(s): 03b778d

Sparse tasks in sentence

Browse files
utilities_language_bert/esp_sentence_bert.py CHANGED
@@ -155,7 +155,7 @@ class SENTENCE:
155
  c_position = 0
156
  bad_target_words = []
157
  for target_word in self.target_words:
158
- position_difference = 3 if target_words_automatic_mode else 0
159
  if not (target_word['position_in_sentence'] == 0
160
  or abs(target_word['position_in_sentence'] - c_position) >= position_difference):
161
  bad_target_words.append(target_word)
 
155
  c_position = 0
156
  bad_target_words = []
157
  for target_word in self.target_words:
158
+ position_difference = 5 if target_words_automatic_mode else 0
159
  if not (target_word['position_in_sentence'] == 0
160
  or abs(target_word['position_in_sentence'] - c_position) >= position_difference):
161
  bad_target_words.append(target_word)
utilities_language_w2v/esp_sentence_w2v.py CHANGED
@@ -202,7 +202,7 @@ class SENTENCE:
202
  c_position = 0
203
  bad_target_words = []
204
  for target_word in self.target_words:
205
- position_difference = 3 if target_words_automatic_mode else 0
206
  if not (target_word['position_in_sentence'] == 0
207
  or abs(target_word['position_in_sentence'] - c_position) >= position_difference):
208
  bad_target_words.append(target_word)
 
202
  c_position = 0
203
  bad_target_words = []
204
  for target_word in self.target_words:
205
+ position_difference = 5 if target_words_automatic_mode else 0
206
  if not (target_word['position_in_sentence'] == 0
207
  or abs(target_word['position_in_sentence'] - c_position) >= position_difference):
208
  bad_target_words.append(target_word)