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-50 /metrics.json
| { | |
| "checkpoint": "checkpoint-50", | |
| "split": "test", | |
| "accuracy": 0.74, | |
| "precision": 0.664, | |
| "recall": 0.74, | |
| "f1_score": 0.6879, | |
| "confusion_matrix": [ | |
| [ | |
| 720, | |
| 57 | |
| ], | |
| [ | |
| 203, | |
| 20 | |
| ] | |
| ] | |
| } |