File size: 874 Bytes
50e9680
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import pandas as pd
import numpy as np
import pickle
import warnings
warnings.filterwarnings("ignore")

def vectorizer():
    category_mapping = {'Debt collection': 0, 
                        'Mortgage': 1, 
                        'Consumer Loan': 2, 
                        'Bank account or service': 3, 
                        'Credit reporting': 4, 
                        'Payday loan': 5, 
                        'Other financial service': 6, 
                        'Student loan': 7, 
                        'Money transfers': 8, 
                        'Prepaid card': 9, 
                        'Credit card': 10
                        }
    
    print("Created Mapping")
    model = pickle.load(open("model.pkl",'rb'))
    vectorizer = pickle.load(open("vectorizer.pkl",'rb'))
    print("Loaded the model")
    return model, vectorizer, category_mapping