bert 将示例转换为特征的问题

2lpgd968  于 5个月前  发布在  其他
关注(0)|答案(2)|浏览(59)

翻译结果为:我有20个类别。尝试使用示例来突出部分特征。

label_list = []
for i in range(20):
    print(i)
    label_list.append(str(i))
print(label_list)
MAX_SEQ_LENGTH = 128
train_features = bert.run_classifier.convert_examples_to_features(train_InputExamples, label_list, MAX_SEQ_LENGTH, tokenizer)

# test_features = bert.run_classifier.convert_examples_to_features(test_InputExamples, label_list, MAX_SEQ_LENGTH, tokenizer)

错误是KeyError,Traceback (most recent call last)
在以下代码中:

775 feature = convert_single_example(ex_index, example, label_list, max_seq_length, tokenizer)
776 features.append(feature)

解决方案:检查label_list中的标签是否正确。

rryofs0p

rryofs0p1#

@abhishek-choudhary-guavus 我遇到了完全相同的问题!你解决了吗?

相关问题