微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Fastai2 如何为DataBlock 编写get_items() 方法

如何解决Fastai2 如何为DataBlock 编写get_items() 方法

我正在努力编写正确的方法get_items()DataBlock。 我正在关注the tutorial

question_label_path = 'text_label.csv'

def getTextLabels(arg):
    df = pd.read_csv(arg)
    return df[['text','label']].to_numpy()
    

textBlock = (TextBlock.from_df(text_cols = ['text'],vocab=dls_lm.vocab),CategoryBlock)

dls_clas = DataBlock(
    blocks=textBlock,get_x = ColReader(cols='text'),get_y = ColReader(cols = "label"),get_items= getTextLabels,splitter = TrainTestSplitter(test_size = 0.2,random_state = 21,stratify=["label"])
).DataLoaders(question_label_path,bs=64)

由于我的工作性质,我无法提供完整的数据集,但它看起来像这样:

enter image description here

错误信息如下:

enter image description here

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。