def label_classes(df, estimated_var): le = LabelEncoder() le.fit(df[estimated_var].values) return le.classes_