terrifyzhao / bert-utils

一行代码使用BERT生成句向量,BERT做文本分类、文本相似度计算

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

运行extract_feature.py

wuaibo opened this issue · comments

有时会报错,ValueError: generator yielded an element of shape (0,) where an element of shape (?, 128) was expected。
另外,输入一句话,每个字都会产生一个词向量,最后要把所有的词向量都想加组成句子的词向量吗?

@wuaibo 句向量是最后对所有字向量做了一个池化操作