Views
No views yet
"all-mpnet-base-v2"XGBRegressorn_estimators=300learning_rate=0.05max_depth=6subsample=0.8colsample_bytree=0.8random_state=421import joblib
2import xgboost as xgb
3from sentence_transformers import SentenceTransformer
4from huggingface_hub import hf_hub_download
5
6# -----------------------------
7# 1. Download model from Hugging Face Hub
8# -----------------------------
9REPO_ID = "mjpsm/Ubuntu_xgb_model" # change if you used a different repo name
10FILENAME = "Ubuntu_xgb_model.pkl"
11
12model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
13
14# -----------------------------
15# 2. Load model + embedder
16# -----------------------------
17model = joblib.load(model_path)
18embedder = SentenceTransformer("all-mpnet-base-v2")
19
20# -----------------------------
21# 3. Example prediction
22# -----------------------------
23text = "During our class project, I made sure everyone’s ideas were included."
24embedding = embedder.encode([text])
25score = model.predict(embedding)[0]
26
27print("Predicted Ubuntu Score:", round(float(score), 3))
28