riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[TextClassificationPipeline] Hotfix: make json serializable

Browse Source
This commit is contained in:
Julien Chaumond 2020-06-17 15:09:22 +00:00
parent 7291ea0bff
commit 70bc3ead4f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines.py
View File

@ -728,7 +728,7 @@ class TextClassificationPipeline(Pipeline):
scores = np.exp(outputs) / np.exp(outputs).sum(-1, keepdims=True)
if self.return_all_scores:
return [
[{"label": self.model.config.id2label[i], "score": score} for i, score in enumerate(item)]
[{"label": self.model.config.id2label[i], "score": score.item()} for i, score in enumerate(item)]
for item in scores
]
else: