在win10环境部署bert-as-service,并使⽤bert⽣成句⼦向量⼩⽩⼀枚,纯属记录⾃⼰的踩雷事件,如有错误,希望可以指出。原创,如有转载,请注明出处。谢谢!
安装完BERT环境后,就⽤pip安装了bert-serving-server和bert-serving-client,bert-as-service⼤致⼯作原理就是运⾏⼀个服务端,然后客户端通过相应的端⼝去和服务端交互,从⽽⽤向量对句⼦进⾏表⽰。
本⽂介绍的是在win10 pycharm上直接运⾏服务端和客户端⽣成句⼦向量的⽅法。详情请看代码:
pycharm安装教程win10下⾯是服务端的代码:(直接新建⼀个service.py,把代码复制进去就可以启动服务)
# -*- coding: utf-8 -*-
from bert_serving.server import BertServer
from bert_serving.server.helper import get_args_parser
def main():
args = get_args_parser().parse_args(['-model_dir', r'C:\project\bert-as-service\chinese_L-12_H-768_A-12',
'-port', '86500',
'-port_out', '86501',
'-max_seq_len', '512',
'-mask_cls_sep',
'-cpu'])
bs = BertServer(args)
bs.start()
if __name__ == "__main__":
main()
下⾯是客户端的代码:
from bert_serving.client import BertClient
bc = BertClient(port=86500, port_out=86501, show_server_config=True, timeout=1000000)
vec = bc.encode(['今天天⽓很好', 'then do it right', 'then do it better'])
print(vec)
直接运⾏就可以的到句⼦向量了。
是,服务端和客户端⽆论谁先启动(若客户端先启动,不应该添加timeout参数),效果是相同的。
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论