property_model / inference.py
rshrott's picture
Create inference.py
4772c07 verified
raw
history blame contribute delete
No virus
250 Bytes
import catboost
from catboost import CatBoostRegressor
model = CatBoostRegressor()
model.load_model("model.cbm")
def predict(features):
# Transform features if necessary
prediction = model.predict([features])
return prediction.tolist()