search

knights-analytics / hugot

Huggingface transformer pipelines in Golang
Apache License 2.0
279 stars 13 forks source link

Support multi-class textClassificationPipeline #11

Closed riccardopinosio closed 6 months ago

riccardopinosio commented 7 months ago

Since we already support textClassification pipelines with a single class label, extending it to multi-class pipelines should be rather straightforward.

riccardopinosio commented 6 months ago

This has been implemented now.