我们如何为ktrain库中的文本分类器使用不同的预训练模型?使用时:
model = text.text_classifier('bert', (x_train, y_train) , preproc=preproc)
This uses the multilangual pretrained model
然而,我也想尝试一个单语模型。即荷兰语:“wietsedv/bert base Dutch cased”,也用于其他k-train实现,for example
但是,当尝试在文本分类器中使用此命令时,它不起作用:
model = text.text_classifier('bert', (x_train, y_train) ,
> preproc=preproc, bert_model='wietsedv/bert-base-dutch-cased')
或
model = text.text_classifier('wietsedv/bert-base-dutch-cased', (x_train, y_train), preproc=preproc)
有人知道怎么做吗?谢谢
ktrain中有两个文本分类API。第一个是
text_classifier
API,可用于选择数量的变压器和非变压器模型。第二个是Transformer
API,它可以与任何transformers
模型一起使用,包括您列出的模型后者在this tutorial notebook和this medium article中详细解释
例如,在下面的示例中,您可以将
MODEL_NAME
替换为您想要的任何模型:例如:
相关问题 更多 >
编程相关推荐