src/transformers/pipelines/text_classification.py · 3f44a66cb617c72efeef0c0b4201cbe2945d8edf · chenpangpang / transformers

"src/routes/vscode:/vscode.git/clone" did not exist on "e6bcdba5ad7d8ba2555392687f5183edf4743733"

Return raw outputs in TextClassificationPipeline (#8328) · 3f44a66c

Lysandre Debut authored Aug 04, 2021



* Return raw outputs in TextClassificationPipeline

* Style

* Support for problem type

* Update src/transformers/pipelines/text_classification.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Apply Nicolas' comments
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

3f44a66c

text_classification.py 6.69 KB

Replace text_classification.py