safeyatra-fastapi / model /ml_model.py
subham-28's picture
Deploying fast api with large graph
53b032e
Raw
History Blame Contribute Delete
318 Bytes
import joblib
import warnings
version ='__0.1.0__'
knn=joblib.load('model/knn_model.pkl')
scaler=joblib.load('model/scaler.pkl')
def predict_score(lat, lon):
warnings.simplefilter(action='ignore', category=UserWarning)
loc = scaler.transform([[lat, lon]])
score=knn.predict(loc)
return score