rubrix/sst2_with_predictions

Comparing model predictions and ground truth labels with Rubrix and Hugging Face Build dataset You can skip this step if you run: from datasets import load_dataset import rubrix as rb ds = rb.DatasetForTextClassification.from_datasets(load_dataset("rubrix/sst2_with_predictions", split="train")) Otherwise, the following cell will run the pipeline over the training set and store labels and predictions. from datasets import load_dataset from transformers import pipeline… See the full description on the dataset page: huggingface.co/datasets/rubrix/sst2_with_predictions.