print result of phoBert
Browse files- phoBERT.py +1 -0
phoBERT.py
CHANGED
|
@@ -78,6 +78,7 @@ def BERT_predict(text):
|
|
| 78 |
token_type_ids = token['token_type_ids']
|
| 79 |
|
| 80 |
result = model(ids, mask, token_type_ids)
|
|
|
|
| 81 |
# print(result)
|
| 82 |
return result.tolist()[0]
|
| 83 |
|
|
|
|
| 78 |
token_type_ids = token['token_type_ids']
|
| 79 |
|
| 80 |
result = model(ids, mask, token_type_ids)
|
| 81 |
+
print(result)
|
| 82 |
# print(result)
|
| 83 |
return result.tolist()[0]
|
| 84 |
|