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

编程问答 2022-04-22

问题描述

我正在努力编写正确的方法：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 (将#修改为@）

data-preprocessingfast-aimachine-learningnlp