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

fix bug

Browse Source
This commit is contained in:
joe 2019-01-31 10:17:30 +08:00
parent 26d9d0238f
commit 6707828af2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extract_feature.py
View File

@ -86,7 +86,7 @@ class BertVector:
def encode(self, sentence):
self.sentence_len = len(sentence)
self.input_queue.put(sentence)
prediction = self.output_queue.get()['encodes'][0]
prediction = self.output_queue.get()['encodes']
return prediction
def queue_predict_input_fn(self):