How to use roncc13/trainCMDBERT-sample with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="roncc13/trainCMDBERT-sample")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("roncc13/trainCMDBERT-sample") model = AutoModelForSequenceClassification.from_pretrained("roncc13/trainCMDBERT-sample", device_map="auto")
loss: 0.04441085457801819
f1: 0.9523809523809523
precision: 0.9090909090909091
recall: 1.0
auc: 1.0
accuracy: 0.95
Files info
Base model
