Skip to content

add support for MultiLabelClassificationModel

User expired requested to merge c7031305/tuhlbox:master into master

I added the parameter "multilabel" to SimpletransformersBertModel, which, when set to True, will use MultiLabelClassificationModel instead of ClassificationModel in the fit method.

Merge request reports