feat: Add multilabel classification for training (#191)
* Added multilabel option to training * Added multilabel option to training * Added multilabel option to training * Added multilabel option to training * Added multilabel option to training * Added multilabel option to training * Added threshold to predict * Updated docs * Updated docs * Removed fallback logic * Updated docs * Updated docs * Resolved feedback * Update model2vec/train/README.md Co-authored-by: Stephan Tulkens <stephantul@gmail.com> * Resolved feedback * Resolved feedback * Resolved feedback * Resolved feedback * add multilabel targets, fix tests (#194) * Fixed bug with array conversion * Optimized inference performance * Changed classes to np array * Added int as possible label type * Added int as possible label type * Use previous logic * Updated type check * Updated type check * Updated type check logic * Only return object type array for multilabel --------- Co-authored-by: Stephan Tulkens <stephantul@gmail.com>
T
Thomas van Dongen committed
2d5151691a1c47129620d9012e82adf8d8e678d6
Parent: 84e3fa8
Committed by GitHub <noreply@github.com>
on 2/16/2025, 1:22:14 PM