Describe the bug
The logged predictions do not show up in annotate mode for textclassification records.
To Reproduce
import rubrix as rb
records = []
for example in to_label1.shuffle():
record = rb.TextClassificationRecord(
inputs=example["text"],
metadata={'category': example['label']}, # log the intents for exploration of specific intents
prediction=[(pred['label'], pred['score']) for pred in example['predictions']],
prediction_agent="distilbert-base-uncased-finetuned-sst-2-english"
)
records.append(record)
rb.log(name='labeling_with_pretrained', records=records)
Basically the first part of our first tutorial.
Expected behavior
Predictions should show up in annotate mode.
Screenshots
Environment (please complete the following information):
Describe the bug The logged predictions do not show up in annotate mode for textclassification records.
To Reproduce
Basically the first part of our first tutorial.
Expected behavior Predictions should show up in annotate mode.
Screenshots
Environment (please complete the following information):
Additional context