BERT 作为分类服务?

Posted

技术标签:

【中文标题】BERT 作为分类服务?【英文标题】:BERT as a service for classification? 【发布时间】:2020-01-19 18:27:36 【问题描述】:

我在分类任务上微调了 BERT。

bert-as-a-service 允许获取词嵌入,但我想获取输入文本的类别。

这个问题最好在这里描述: https://github.com/hanxiao/bert-as-service/issues/213

关于如何做到这一点的任何想法?

【问题讨论】:

使用 Bert 作为服务进行特征化,然后用它们构建一个简单的分类器 谢谢!如果我有一个训练有素的模型,是否需要再次训练分类器? 意思是,如果你从 BERT-as-Service (BaS) 输出训练分类器,那么你改变 BaS 使用的 BERT 模式,是否需要重新训练分类器?如果这就是你的意思,那么是的 嗨,不。 Bert as a service 提供了一个向量而不是类。我有一个微调模型 Weaviate有语义分类功能:semi.technology/documentation/weaviate/current/classification/… 【参考方案1】:

Bert-as-a-Service 的目的是生成类似于 ELMo 的句子嵌入,而不是用作分类器。

要在微调后使用 BERT 进行分类,可以使用以下链接中给出的实现:

https://github.com/google-research/bert/blob/master/predicting_movie_reviews_with_bert_on_tf_hub.ipynb

【讨论】:

以上是关于BERT 作为分类服务?的主要内容,如果未能解决你的问题,请参考以下文章

自然语言处理动手学Bert文本分类

BERT模型在多类别文本分类时的precision, recall, f1值的计算

如何标准化 BERT 分类器的输出

使用预先训练的BERT模型对多类文本分类进行错误分类

将 LIME 解释应用于我的微调 BERT 以进行序列分类模型?

为啥 Bert 转换器使用 [CLS] 令牌进行分类,而不是对所有令牌进行平均?