在 imdb 电影评论数据集上训练罗伯塔模型给出这个错误？

问题描述

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,axis = 1,)

  
    return train_InputExamples,validation_InputExamples

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

NameError：未定义名称“InputExample”

运行这部分代码后，它给出了错误。请告诉我如何解决这个错误。

解决方法

from transformers import InputExample