python-3.x - 为 Flair 模型训练准备数据框

Question

我正在尝试将以下格式的数据帧获取到 txt 文件中，如屏幕截图所示。

我与显示的代码比较接近，但当我尝试将 txt 读入我的 ML 算法时，出现以下错误:“labels = set([label.value for sent in self.train for label in sent.labels] )AttributeError:“NoneType”对象没有属性“标签””

有人可以帮忙吗？

output['label']=['__label__'+ s for s in output['label'].astype(str)]
output['text']= output['text'].replace('\n',' ', regex=True).replace('\t',' ', regex=True)
output.to_csv(r'sst_train.csv', index=False, sep=' ', header=False)

Answer 1

以下似乎有效:

lab = data['label'].tolist()
text = data['text'].tolist()
with open('test.txt', 'w') as outfile:
    for x in range(len(lab)):
        l1 = str(lab[x])
        print(type(l1))
        t1 = str(text[x])
        print(t1)
        d = 'ewdwed'
        message = "__label__{}\t{}\n".format(l1,t1)
        outfile.write(message)