Views
No views yet
0) or buggy (label 1).0 = clean code1 = buggy / defective code"func": raw function-level source code"target": binary label (0 = clean, 1 = buggy)DataCollatorWithPadding| Hyperparameter | Value |
|---|---|
| Epochs | 1 |
| Learning rate | 2e-5 |
| Batch size | 8 |
| FP16 | Yes |
| Max length | 512 |
| Optimizer | AdamW |
| Loss | Cross-entropy |
| remove_unused_columns | False |
1model = AutoModelForSequenceClassification.from_pretrained(
2 "microsoft/deberta-v3-base", num_labels=2
3)
4
5TrainingArguments(
6 output_dir="filter_model",
7 learning_rate=2e-5,
8 num_train_epochs=1,
9 fp16=True,
10 per_device_train_batch_size=8,
11 logging_steps=50,
12 remove_unused_columns=False,
13)