Code Task Classifier (TF-IDF + LightGBM)
Distilled from Qwen2.5-Coder pseudo-labels (26 programming task categories).
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)))
- Downloads last month
- -
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("elzahar05/code-task-classifier-lgbm", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html