def get_sizes(self, train):
    class_data = peek(train)
    row = class_data[0]
    penultimates = get_penultimates(list(row['features'].keys()))
    return [row['features'][k].shape[(- 1)] for k in penultimates]
def get_sizes(self, train):
    class_data = peek(train)
    row = class_data[0]
    penultimates = get_penultimates(list(row['features'].keys()))
    return [row['features'][k].shape[(- 1)] for k in penultimates]