File size: 388 Bytes
cf86603
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from sklearn.preprocessing import LabelEncoder
import pickle

# These are the categories you need for your app
categories = ['PAYMENT', 'TRANSFER', 'CASH_OUT', 'DEBIT', 'CASH_IN', 'Yes', 'No']

# Create the LabelEncoder and fit it
le = LabelEncoder()
le.fit(categories)

# Save the LabelEncoder to a new pickle file
with open('label_encoder.pkl', 'wb') as file:
    pickle.dump(le, file)