sileod / tasknet

Easy multi-task learning with HuggingFace Datasets and Trainer
GNU General Public License v3.0
40 stars 3 forks source link

Return all scores from a text classification pipeline #4

Closed tboquet closed 10 months ago

tboquet commented 10 months ago

Currently this is not a parameter that we can pass to the TextClassificationPipeline from the load_pipeline function. It would be interesting to have the functionality when using the model on specific tasks for inference.