def get_feature_names(self, train):
    class_data = peek(train)
    row = peek(class_data)
    return sorted(list(row.keys()))
def get_feature_names(self, train):
    class_data = peek(train)
    row = peek(class_data)
    return sorted(list(row.keys()))