StanfordSCALE/assertions_llm_annotated_talkmoves
Viewer • Updated • 6.43k • 38
How to use StanfordSCALE/assertion_sentence_has_apology with setfit:
from setfit import SetFitModel
model = SetFitModel.from_pretrained("StanfordSCALE/assertion_sentence_has_apology")This classifier was trained for EduBehaviors: Assertion-based schemas for auditable dialogue coding and is usable through the Python package EduBehaviors-kit. This classifier was trained on an LLM-annotated subset of teacher utterances from the TalkMoves Dataset. See the Datasets section below for more information.
| Dataset | Split | Size |
|---|---|---|
| StanfordSCALE/assertions_llm_annotated_talkmoves | train | 3,432 (53.3%) |
| StanfordSCALE/assertions_llm_annotated_talkmoves | dev | 858 (13.3%) |
| StanfordSCALE/assertions_llm_annotated_talkmoves | test | 2,144 (33.3%) |
This model's columns are assertion_sentence_has_apology and split_sentence_has_apology.
Base rate (share of rows labeled as True): 0.4% overall — 0.4% train, 0.2% dev, 0.3% test.
Labels were generated with LLM annotators. Krippendorff's alpha for this assertion is 0.869.
| Parameter | Value |
|---|---|
| Base model (body) | sentence-transformers/paraphrase-mpnet-base-v2 |
| Head | LogisticRegression |
| Body learning rate | 2e-05 |
| Head learning rate | 0.01 |
| Batch size | 16 (contrastive phase) / 32 (head) |
| Epochs | 10 |
| Max steps | 5000 (contrastive phase) |
| Eval max steps | 100 |
| Seed | 20260904 |
| Mixed precision | enabled on GPU |
| Split | n | Base rate | Precision | Recall | F1 (positive class) | ROC-AUC | Average precision |
|---|---|---|---|---|---|---|---|
| dev | 858 | 0.2% | 1.000 | 1.000 | 1.000 | 1.000 | 1.000 |
| test | 2,144 | 0.3% | 1.000 | 0.667 | 0.800 | 1.000 | 1.000 |
The model was trained on text built as:
{utterance}
The utterance is passed through as-is.
pip install setfit
from setfit import SetFitModel
model = SetFitModel.from_pretrained("StanfordSCALE/assertion_sentence_has_apology")
text = 'I apologize but I keep doing it'
model.predict([text]) # -> array([1]) when the assertion holds
model.predict_proba([text]) # -> [[P(no), P(yes)]]
@misc{assertion_sentence_has_apology,
author = {Stanford SCALE Initiative},
title = {Assertion classifier: sentence has apology},
year = {2026},
url = {https://huggingface.co/StanfordSCALE/assertion_sentence_has_apology}
}