Upload BertForJointParsing.py

BertForJointParsing.py

@@ -431,7 +431,7 @@ def convert_output_to_ud(output_sentences, htb_extras=False):
         for start,end,token in ranges:
             if end - start > 1:
                 cur_output.append(f'{start + 1}-{end}\t{token}\t_\t_\t_\t_\t_\t_\t_\t_')
-            for idx,output in enumerate(intermediate_output[start:end], start):
+            for idx,output in enumerate(intermediate_output[start:end], start + 1):
                 # compute the actual dependency location
                 dep = output['dep'] if output.get('absolute_dep', False) else idx_to_key[output['dep']]
                 # and add the full ud string in