def print_label_dict(self, data_df):
    for item in self.label_dict.values():
        print(item, data_df.loc[(data_df.Y == item), :].shape[0])
    print('==========')
def print_label_dict(self, data_df):
    for item in self.label_dict.values():
        print(item, data_df.loc[(data_df.Y == item), :].shape[0])
    print('==========')