albert 中文使用

最近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)

調試結果如下：