Advertisement
Guest User

Untitled

a guest
Mar 27th, 2019
103
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.85 KB | None | 0 0
  1. scoring.py
  2. ============
  3.  
  4. import pickle
  5. import psycopg2
  6. import pandas as pd
  7.  
  8. from settings import (
  9.     DB_NAME, DB_USER, DB_PASSWORD, DB_PORT
  10. )
  11.  
  12.  
  13. def select_params_from_db(app_id):
  14.     sql = """
  15.        SeLECT *
  16.        from app_params
  17.        where id=123
  18.    """
  19.     con = psycopg2.connect(
  20.         database=DB_NAME,
  21.         user=DB_USER,
  22.         password=DB_PASSWORD,
  23.         host=DB_PORT
  24.     )
  25.     df = pd.read_sql(sql, con)
  26.     print(df)
  27.     return df
  28.  
  29.  
  30. def call_model(params):
  31.     pkl_model_name = 'c:/score/test_model.pkl'
  32.     with open(pkl_model_name, 'rb') as file:
  33.         pickle_model = pickle.load(file)
  34.         score = pickle_model.predict_proba(params)
  35.         print(score)
  36.         return score
  37.  
  38.  
  39. def get_score(app_id):
  40.     params = select_params_from_db(app_id)
  41.     score = call_model(params)
  42.     return score
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement