Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- scoring.py
- ============
- import pickle
- import psycopg2
- import pandas as pd
- from settings import (
- DB_NAME, DB_USER, DB_PASSWORD, DB_PORT
- )
- def select_params_from_db(app_id):
- sql = """
- SeLECT *
- from app_params
- where id=123
- """
- con = psycopg2.connect(
- database=DB_NAME,
- user=DB_USER,
- password=DB_PASSWORD,
- host=DB_PORT
- )
- df = pd.read_sql(sql, con)
- print(df)
- return df
- def call_model(params):
- pkl_model_name = 'c:/score/test_model.pkl'
- with open(pkl_model_name, 'rb') as file:
- pickle_model = pickle.load(file)
- score = pickle_model.predict_proba(params)
- print(score)
- return score
- def get_score(app_id):
- params = select_params_from_db(app_id)
- score = call_model(params)
- return score
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement