Text classification on a custom dataset using Huggingface Trainerš¤
By defining the TrainingArguments and instantiating the Trainer, we can create a model to fine-tune DistilBert on the prepared dataset. As part of fine-tuning with minimal changes to the DistilBert, we will be training the classifier(Feed-forward neural network softmax).
Nov-28-2022, 07:35:06 GMT
- Technology: