-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
30 lines (18 loc) · 1.39 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
VERSION = '0.1.0'
DATASET = "loan_final313.csv"
PATH_TO_DATASET = 'data/'
ACCEPTABLE_COLUMNS = ['id', 'year', 'issue_d', 'final_d', 'emp_length_int', 'home_ownership',
'home_ownership_cat', 'income_category', 'annual_inc', 'income_cat',
'loan_amount', 'term', 'term_cat', 'application_type',
'application_type_cat', 'purpose', 'purpose_cat', 'interest_payments',
'interest_payment_cat', 'loan_condition', 'loan_condition_cat',
'interest_rate', 'grade', 'grade_cat', 'dti', 'total_pymnt',
'total_rec_prncp', 'recoveries', 'installment', 'region']
CATEGORICAL_COLUMNS = ['issue_d', 'home_ownership', 'income_category', 'term', 'application_type', 'purpose', 'interest_payments', 'grade', 'region']
NUMERICAL_COLUMNS = ['id', 'year', 'final_d', 'emp_length_int', 'home_ownership_cat', 'annual_inc', 'income_cat', 'loan_amount', 'term_cat', 'application_type_cat', 'purpose_cat', 'interest_payment_cat', 'loan_condition', 'loan_condition_cat', 'interest_rate', 'grade_cat', 'dti', 'total_pymnt', 'total_rec_prncp', 'recoveries', 'installment']
DROP_THIS = ['id']
UNNECESSARY_COLUMNS = ['year','issue_d','home_ownership_cat','income_cat','term_cat','application_type_cat','purpose_cat','interest_payment_cat','loan_condition_cat','grade_cat']
TARGET_COLUMN = 'loan_condition'
# edit this according your url
URL = 'http://127.0.0.1:5000'
HEROKU_URL = 'https://cst-model-deployment.herokuapp.com'