Skip to content
GitLab

add support for MultiLabelClassificationModel

Open Manfred Moosleitner 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.