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-100 /metrics.json
| { | |
| "checkpoint": "checkpoint-100", | |
| "split": "test", | |
| "accuracy": 0.734, | |
| "precision": 0.6704, | |
| "recall": 0.734, | |
| "f1_score": 0.6916, | |
| "confusion_matrix": [ | |
| [ | |
| 707, | |
| 70 | |
| ], | |
| [ | |
| 196, | |
| 27 | |
| ] | |
| ] | |
| } |