Instructions to use dzungpham/graphcodebert-code-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dzungpham/graphcodebert-code-classification with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("dzungpham/graphcodebert-code-classification", dtype="auto") - Notebooks
- Google Colab
- Kaggle
graphcodebert-code-classification / inference /graphcodebert-mixcode-fft /checkpoint-200 /metrics.json
| { | |
| "checkpoint": "checkpoint-200", | |
| "split": "test", | |
| "accuracy": 0.588, | |
| "precision": 0.6758, | |
| "recall": 0.588, | |
| "f1_score": 0.6189, | |
| "confusion_matrix": [ | |
| [ | |
| 490, | |
| 287 | |
| ], | |
| [ | |
| 125, | |
| 98 | |
| ] | |
| ] | |
| } |