How to use arpanghoshal/EkmanClassifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="arpanghoshal/EkmanClassifier")
# Load model directly from transformers import AutoTokenizer, BertForMultiLabelClassification tokenizer = AutoTokenizer.from_pretrained("arpanghoshal/EkmanClassifier") model = BertForMultiLabelClassification.from_pretrained("arpanghoshal/EkmanClassifier")
Hi!
If possible, could you please include your achieved metrics and the hyperparams you used to fine-tune BERT in your README.md?Also, I see that you used the Go dataset. Did you just merge the 28 classes into these 6?
README.md
Thanks a lot!
· Sign up or log in to comment
