emtropyml

Support for multi-label and multi-class text classification using DistilBERT

2

How can I use DistilBERT for multi-label classification for building a fast and deploy-able model?

AssertionError: Bad argument number for Name: 3, expecting 4

15

--> model = MultiLabelClassifier(base_model=DistilBERT, batch_size=2, multi_label_sequences=True, n_epochs=3) --> model.fit(trainX, trainY) WARNING: Entity could not be transformed and will be executed as-is. Please report this to the AutgoGraph team. When filing...