search

mrseanryan / csfy

Train a simple text classifier and predict labels - supports ONNX output for performance, language-neutral
https://medium.com/p/9fd7003000fc
MIT License
0 stars 0 forks source link

Add support for multi-class (multi label) training #6

Open mrseanryan opened 3 months ago

mrseanryan commented 3 months ago

Possibly would need separate model - since expecting to train to a single 'label' per input:

https://sbert.net/docs/sentence_transformer/training_overview.html

but could try a trick, where the input is 'tagged' like: "[] text" => "expected-label-1-value" "[] text" => "expected-label-2-value"