File size: 490 Bytes
c97fe10 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from joblib import load
from datetime import datetime, timedelta
model = load("amc_model.pkl")
def predict_next_service(last_service_date):
try:
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")
except Exception as e:
raise ValueError(f"Prediction error: {str(e)}") |