Ramdevs_Fraud_Detector / label_encoder.pkl
kkhushisaid's picture
Create label_encoder.pkl
cf86603 verified
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)