Как изменить значения меток в файле jsonl на значения из списка прогнозов

У меня есть такой файл jsonl и список прогнозов модели машинного обучения. Как изменить значения в метке на значения из прогнозов? Вот моя функция обработки, но ничего не происходит. Как я могу изменить свою функцию, чтобы заменить или написать новую функцию, а затем сохранить все в формате jsonl?


print(predictions) ## [0, 1, 0...0, 0, 0]

def get_answer(dic):
  for i in int_predictions:
    dic['label'] = 0
  return dic
import random
for i in tqdm(range(len(new_test))):
    q = new_test[i]['passage']['questions']
    for j in range(len(q)):
        #print(q[j])
        q[j]['answers'] = [get_answer(dic) for dic in q[j]['answers'] ]
    new_test[i]['passage']['questions'] = q

{'idx': 0,
 'passage': {'questions': [{'answers': [{'idx': 0, 'label': 0},
     {'idx': 1, 'label': 1},
     {'idx': 2, 'label': 1},
     {'idx': 3, 'label': 1}],
    'idx': 0},
   {'answers': [{'idx': 4, 'label': 1},
     {'idx': 5, 'label': 1},
     {'idx': 6, 'label': 1},
     {'idx': 7, 'label': 1}],
    'idx': 1},
   {'answers': [{'idx': 8, 'label': 1},
     {'idx': 9, 'label': 1},
     {'idx': 10, 'label': 1},
     {'idx': 11, 'label': 1}],
    'idx': 2},
   {'answers': [{'idx': 12, 'label': 1},
     {'idx': 13, 'label': 1},
     {'idx': 14, 'label': 1},
     {'idx': 15, 'label': 1}],
    'idx': 3},
   {'answers': [{'idx': 16, 'label': 1},
     {'idx': 17, 'label': 1},
     {'idx': 18, 'label': 1},
     {'idx': 19, 'label': 1},
     {'idx': 20, 'label': 1}],
    'idx': 4},
   {'answers': [{'idx': 21, 'label': 1},
     {'idx': 22, 'label': 1},
     {'idx': 23, 'label': 1},
     {'idx': 24, 'label': 1},
     {'idx': 25, 'label': 1}],
    'idx': 5},
   {'answers': [{'idx': 26, 'label': 1},
     {'idx': 27, 'label': 1},
     {'idx': 28, 'label': 1},
     {'idx': 29, 'label': 1},
     {'idx': 30, 'label': 1}],
    'idx': 6},
   {'answers': [{'idx': 31, 'label': 1},
     {'idx': 32, 'label': 1},
     {'idx': 33, 'label': 1},
     {'idx': 34, 'label': 1},
     {'idx': 35, 'label': 1}],
    'idx': 7},
   {'answers': [{'idx': 36, 'label': 1},
     {'idx': 37, 'label': 1},
     {'idx': 38, 'label': 1},
     {'idx': 39, 'label': 1},
     {'idx': 40, 'label': 1}],
    'idx': 8},
   {'answers': [{'idx': 41, 'label': 1},
     {'idx': 42, 'label': 1},
     {'idx': 43, 'label': 1},
     {'idx': 44, 'label': 1},
     {'idx': 45, 'label': 1}],
    'idx': 9}]}}

Ответы (0 шт):