StanfordSCALE/assertions_llm_annotated_talkmoves
Viewer • Updated • 24.7k • 98
How to use StanfordSCALE/assertion_grants_permission with setfit:
from setfit import SetFitModel
model = SetFitModel.from_pretrained("StanfordSCALE/assertion_grants_permission")
preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"])
print(preds)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 | 13,182 (53.3%) |
| StanfordSCALE/assertions_llm_annotated_talkmoves | dev | 3,296 (13.3%) |
| StanfordSCALE/assertions_llm_annotated_talkmoves | test | 8,237 (33.3%) |
This model's columns are assertion_grants_permission and split_grants_permission.
Base rate (share of rows labeled as True): 1.7% overall — 1.7% train, 1.4% dev, 1.9% test.
Labels were generated with LLM annotators. Krippendorff's alpha for this assertion is 0.739.
| 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 | 3,296 | 1.4% | 0.688 | 0.478 | 0.564 | 0.847 | 0.467 |
| test | 8,237 | 1.9% | 0.589 | 0.474 | 0.525 | 0.925 | 0.497 |
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_grants_permission")
text = 'If youre not sure you can talk to your partner'
model.predict([text]) # -> array([1]) when the assertion holds
model.predict_proba([text]) # -> [[P(no), P(yes)]]
@misc{assertion_grants_permission,
author = {Stanford SCALE Initiative},
title = {Assertion classifier: grants permission},
year = {2026},
url = {https://huggingface.co/StanfordSCALE/assertion_grants_permission}
}