AMCPREDICTAPP / model_predictor.py
Abhi10000's picture
Create model_predictor.py
e66fbad verified
raw
history blame contribute delete
384 Bytes
from joblib import load
from datetime import datetime, timedelta
model = load("amc_model.pkl")
def predict_next_service(last_service_date):
today = datetime.today()
delta_days = (today - datetime.strptime(last_service_date, "%Y-%m-%d")).days
predicted_days = int(model.predict([[delta_days]])[0])
return (today + timedelta(days=predicted_days)).strftime("%Y-%m-%d")