Views
No views yet
all-mpnet-base-v2)0.0 – 1.00.008660.8920.1 away from true labels on the 0–1 scale.1import xgboost as xgb
2from sentence_transformers import SentenceTransformer
3from huggingface_hub import hf_hub_download
4
5# -----------------------------
6# 1. Download model from Hugging Face
7# -----------------------------
8REPO_ID = "mjpsm/Imani-xgb-model"
9FILENAME = "Imani_xgb_model.json"
10
11model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
12
13# -----------------------------
14# 2. Load Model + Embedder
15# -----------------------------
16model = xgb.XGBRegressor()
17model.load_model(model_path)
18
19embedder = SentenceTransformer("all-mpnet-base-v2")
20
21# -----------------------------
22# 3. Example Prediction
23# -----------------------------
24text = "I reminded my cousin that storms always pass."
25embedding = embedder.encode([text])
26score = model.predict(embedding)[0]
27
28print("Predicted Imani Score:", round(float(score), 3))