Team Ai
Modelpublic

StanfordSCALE/assertion_sentence_has_comparison_terms

sourceHugging Faceupdated 22d agoView on Hugging Face
0likes35downloads
Model Card

Assertion: sentence has comparison terms

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.


Training Details

Datasets

This model's columns are assertion_sentence_has_comparison_terms and split_sentence_has_comparison_terms.

Base rate (share of rows labeled as True): 6.2% overall — 6.2% train, 6.3% dev, 6.2% test.

Labels and annotation

Labels were generated with LLM annotators. Krippendorff's alpha for this assertion is 0.657.

Hyperparameters

ParameterValue
Base model (body)sentence-transformers/paraphrase-mpnet-base-v2
HeadLogisticRegression
Body learning rate2e-05
Head learning rate0.01
Batch size16 (contrastive phase) / 32 (head)
Epochs10
Max steps5000 (contrastive phase)
Eval max steps100
Seed20260904
Mixed precisionenabled on GPU

Evaluation

Results

SplitnBase ratePrecisionRecallF1 (positive class)ROC-AUCAverage precision
dev8586.3%0.6770.7780.7240.9620.748
test2,1466.2%0.7790.8130.7960.9430.840

Limitations

  • —Labels come from LLM annotators, not human coders. Agreement between annotators with Krippendorff's Alpha is 0.657.
  • —Trained on teacher utterances only. Behaviour on student speech is untested.

How to Use

Message Structure

The model was trained on text built as:

{utterance}

The utterance is passed through as-is.

Running instructions

bash
pip install setfit
python
from setfit import SetFitModel

model = SetFitModel.from_pretrained("StanfordSCALE/assertion_sentence_has_comparison_terms")

text = 'What I want you to focus on today is how can you relate the two diameters to the slant height and then how can you kind of think about the relationship between all three measurements'
model.predict([text])        # -> array([1]) when the assertion holds
model.predict_proba([text])  # -> [[P(no), P(yes)]]

Citation

bibtex
@misc{assertion_sentence_has_comparison_terms,
  author = {Stanford SCALE Initiative},
  title  = {Assertion classifier: sentence has comparison terms},
  year   = {2026},
  url    = {https://huggingface.co/StanfordSCALE/assertion_sentence_has_comparison_terms}
}