Skip to content

add support for MultiLabelClassificationModel

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

Loading