multilabel-classification-bert / configuration_multilabelbert.py
jvaquet's picture
Upload BertForMultiLabelTokenClassification
5bb7ef5 verified
raw
history blame contribute delete
230 Bytes
from transformers import BertConfig, AutoConfig
class MultiLabelBertConfig(BertConfig):
model_type = 'MultiLabelBert'
AutoConfig.register('MultiLabelBert', MultiLabelBertConfig)
MultiLabelBertConfig.register_for_auto_class()