How to use namesarnav/causalbench_code-bert-base-uncased with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="namesarnav/causalbench_code-bert-base-uncased")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("namesarnav/causalbench_code-bert-base-uncased") model = AutoModelForSequenceClassification.from_pretrained("namesarnav/causalbench_code-bert-base-uncased", device_map="auto")
f834ad9 248b1fe f834ad9
1
2
3
4
5
6
7
8
9
{ "eval_loss": 0.5690733194351196, "eval_accuracy": 0.7080152671755725, "eval_macro_f1": 0.7029638652718904, "eval_runtime": 258.0623, "eval_samples_per_second": 6.092, "eval_steps_per_second": 0.194, "epoch": 5.0 }
