albert 中文使用

最近尝试了 一种更简单使用 albert 和 Bert的方法,具体 url 如下

https://blog.csdn.net/renyuanfang/article/details/106556660

最近Google 设计了一个【精简版的BERT】ALBERT,预测性能和bert相比大大提高,自己尝试直接调用其开源代码:https://github.com/google-research/ALBERT 未能直接跑通,自己调试成功的代码地址如下:https://github.com/HappyYuanFang/albert_demo.git ,该代码可直接在windows  pycharm中跑通

使用方式如下参考 hello_bert.py

from albert.extract_feature import BertVector
pooling_strategy = "REDUCE_MEAN"
# pooling_strategy = "NONE"
bc = BertVector(pooling_strategy=pooling_strategy, max_seq_len=80)
s1 = '谢谢你,每一个平凡的中国人'
v = bc.encode([s1])
v1 = v["encodes"][0]
print(v1)

调试结果如下: 

albert 中文使用_第1张图片

你可能感兴趣的:(machine,learning)