参考https://github.com/hanxiao/bert-as-service
一.第一步,安装python包
pip install bert-serving-server # server
pip install bert-serving-client # client, independent of `bert-serving-server`
二.下载BERT预训练模型
访问https://github.com/google-research/bert,来到
这里推荐下载的是BERT-Base,Uncased和BERT-Base,Chinese。因为我要做中文词向量的查询,所以下载了Chinese这版的。
下载完后解压,拷贝一下文件绝对路径(我的存放位置是F:/MyDownloads/chinese_L-12_H-768_A-12)
三.打开cmd,运行命令行
bert-serving-start -model_dir F:/MyDownloads/chinese_L-12_H-768_A-12 -num_worker=1
这里的model_dir是BERT预训练模型所在路径,num_worker为同时访问限制,一般自己本地用的话1就可以了。
四.词向量查询
from bert_serving.client
import BertClient
bc = BertClient()
print(bc.encode(['中国', '美国', '日本']))