How to use k-habib/scram-model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="k-habib/scram-model")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("k-habib/scram-model") model = AutoModelForSequenceClassification.from_pretrained("k-habib/scram-model", device_map="auto")
loss: 0.08497993648052216
f1: 0.9870281300102026
precision: 0.9791787160208213
recall: 0.995004407875404
auc: 0.9946037165171522
accuracy: 0.984038737446198
Files info
Base model
