NLP/bert-utils
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改验证集

Browse Source
This commit is contained in:
joe 2019-01-30 10:39:56 +08:00
parent 0ca26576a4
commit 13c1691426
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -1,6 +1,6 @@
# bert-utils
本文基于Google开源的[BERT]()代码进行了进一步的简化，方便生成句向量与做文本分类
本文基于Google开源的[BERT](https://github.com/google-research/bert)代码进行了进一步的简化，方便生成句向量与做文本分类
1、下载BERT中文模型

0
data/test.csv → data/dev.csv
View File

Can't render this file because it is too large.

20
similarity.py
View File

@ -664,13 +664,13 @@ class BertSim:
if __name__ == '__main__':
sim = BertSim()
# sim.set_mode(tf.estimator.ModeKeys.TRAIN)
# sim.train()
# sim.set_mode(tf.estimator.ModeKeys.EVAL)
# sim.eval()
sim.set_mode(tf.estimator.ModeKeys.PREDICT)
while True:
sentence1 = input('sentence1: ')
sentence2 = input('sentence2: ')
predict = sim.predict(sentence1, sentence2)
print(f'similarity：{predict[0][1]}')
sim.set_mode(tf.estimator.ModeKeys.TRAIN)
sim.train()
sim.set_mode(tf.estimator.ModeKeys.EVAL)
sim.eval()
# sim.set_mode(tf.estimator.ModeKeys.PREDICT)
# while True:
# sentence1 = input('sentence1: ')
# sentence2 = input('sentence2: ')
# predict = sim.predict(sentence1, sentence2)
# print(f'similarity：{predict[0][1]}')