def get_minimal_dataset(df, output_column='subreddit'):
    """Return a dataset that has one example per output class."""
    return df.groupby(output_column, group_keys=False).apply(lambda df: df.head(1))