transformers.js
cc4b857d - Add problem type (Fixes #248) (#249)

Commit

2 years ago

Add problem type (Fixes #248) (#249) * Add support for `problem_type` in text classification * Add unit test for `multi_label_classification` problem type * Update supported_models.py

References

#249 - Add problem type (Fixes #248)

Author

xenova

Parents

d7a73434

Files3

scripts
- supported_models.py
src
- pipelines.js
tests
- pipelines.test.js

transformers.js cc4b857d - Add problem type (Fixes #248) (#249)

transformers.js
cc4b857d - Add problem type (Fixes #248) (#249)