| library_name: sklearn | |
| tags: [text-classification, lightgbm, tfidf, code] | |
| # Code Task Classifier (TF-IDF + LightGBM) | |
| Distilled from Qwen2.5-Coder pseudo-labels (26 programming task categories). | |
| ```python | |
| import joblib | |
| from huggingface_hub import hf_hub_download | |
| path = hf_hub_download("elzahar05/code-task-classifier-lgbm", "student_lgbm_model.joblib") | |
| art = joblib.load(path) | |
| X = art["vectorizer"].transform(["Write a function to reverse a linked list"]) | |
| print(art["label_encoder"].inverse_transform(art["model"].predict(X))) | |
| ``` | |