本地安装bert服务，并查询词向量

参考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(['中国', '美国', '日本']))