Spaces:
Sleeping
Sleeping
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)
|