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)