elzahar05 commited on
Commit
8ded19f
·
verified ·
1 Parent(s): 5cf0acb

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: sklearn
3
+ tags: [text-classification, lightgbm, tfidf, code]
4
+ ---
5
+ # Code Task Classifier (TF-IDF + LightGBM)
6
+
7
+ Distilled from Qwen2.5-Coder pseudo-labels (26 programming task categories).
8
+
9
+ ```python
10
+ import joblib
11
+ from huggingface_hub import hf_hub_download
12
+
13
+ path = hf_hub_download("elzahar05/code-task-classifier-lgbm", "student_lgbm_model.joblib")
14
+ art = joblib.load(path)
15
+ X = art["vectorizer"].transform(["Write a function to reverse a linked list"])
16
+ print(art["label_encoder"].inverse_transform(art["model"].predict(X)))
17
+ ```