File size: 561 Bytes
8e3bbfe a0f3d24 b4fadea 8e3bbfe 944b0c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | # app/inference/predictor.py
# model.predict wrapper
import json
import joblib
import numpy as np
from app.core.config import MODEL_PATH, FEATURES_PATH
class Predictor:
def __init__(self):
self.model = joblib.load(MODEL_PATH)
with open(FEATURES_PATH, "r") as f:
self.features = json.load(f)
self.model_version = "v1"
def predict(self, df):
X = df[self.features]
probas = self.model.predict_proba(X)[:, 1]
preds = (probas >= 0.5).astype(int)
return preds.tolist(), probas.tolist()
|