AMCPREDICTAPP / train_model.py
Abhi10000's picture
Create train_model.py
ebf1190 verified
raw
history blame contribute delete
263 Bytes
import pandas as pd
from sklearn.linear_model import LinearRegression
from joblib import dump
data = pd.read_csv("training_data.csv")
X = data[["last_service_days"]]
y = data["next_service_days"]
model = LinearRegression().fit(X, y)
dump(model, "amc_model.pkl")