BERT作为分类服务?

时间:2019-09-18 18:26:47

标签: text nlp classification

我对分类任务进行了微调。

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

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

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

1 个答案:

答案 0 :(得分:0)

“即服务即服务”的目的是产生类似于ELMo的句子嵌入,而不是用作分类器。

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

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