How to use schematise/ICAT-text-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="schematise/ICAT-text-classifier")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("schematise/ICAT-text-classifier") model = AutoModelForSequenceClassification.from_pretrained("schematise/ICAT-text-classifier")
loss: 0.20267261564731598
f1: 0.9090909090909091
precision: 0.8333333333333334
recall: 1.0
auc: 0.9666666666666667
accuracy: 0.9411764705882353
Files info
