Обучение модели роберты на наборе данных обзоров фильмов imdb дает эту ошибку?

Question

       def convert_data_to_examples(train, test, review, sentiment): 
train_InputExamples = train.apply(lambda x: InputExample(guid=None, # Globally unique ID for bookkeeping, unused in this case
                                                      text_a = x[review], 
                                                      label = x[sentiment]), axis = 1)

validation_InputExamples = test.apply(lambda x: InputExample(guid=None, # Globally unique ID for bookkeeping, unused in this case
                                                      text_a = x[review], 
                                                      label = x[sentiment]), axis = 1,)


return train_InputExamples, validation_InputExamples
train_InputExamples, validation_InputExamples = convert_data_to_examples(train,  test, 'review',  'sentiment')

NameError: имя InputExample не определено

После запуска этой части кода возникает ошибка. Подскажите пожалуйста, как решить эту ошибку.

Answer 1

      from transformers import InputExample