xanderbeberman's picture
publish assertion encoder with model card
9d398c0 verified
|
Raw History Blame Contribute Delete
4.09 kB
metadata
language: en
library_name: setfit
pipeline_tag: text-classification
base_model: sentence-transformers/paraphrase-mpnet-base-v2
datasets:
  - StanfordSCALE/assertions_llm_annotated_talkmoves
metrics:
  - f1
  - precision
  - recall
  - roc_auc
tags:
  - setfit
  - text-classification
  - education
  - classroom-discourse
  - edubehaviors
  - assertion
model-index:
  - name: assertion_sentence_has_time_reference
    results:
      - task:
          type: text-classification
        dataset:
          name: assertions_llm_annotated_talkmoves
          type: StanfordSCALE/assertions_llm_annotated_talkmoves
          split: test
        metrics:
          - type: f1
            value: 0.728
            name: F1 (positive class, test)
          - type: precision
            value: 0.7154
            name: Precision (positive class, test)
          - type: recall
            value: 0.741
            name: Recall (positive class, test)
          - type: roc_auc
            value: 0.9241
            name: ROC-AUC (test)

Assertion: sentence has time reference

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_time_reference and split_sentence_has_time_reference.

Base rate (share of rows labeled as True): 17.3% overall — 17.2% train, 19.2% dev, 16.9% test.

Labels and annotation

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

Hyperparameters

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

Evaluation

Results

Split n Base rate Precision Recall F1 (positive class) ROC-AUC Average precision
dev 856 19.2% 0.687 0.695 0.691 0.888 0.704
test 2,146 16.9% 0.715 0.741 0.728 0.924 0.792

Limitations

  • Labels come from LLM annotators, not human coders. Agreement between annotators with Krippendorff's Alpha is 0.603.
  • 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

pip install setfit
from setfit import SetFitModel

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

text = 'Happy Friday'
model.predict([text])        # -> array([1]) when the assertion holds
model.predict_proba([text])  # -> [[P(no), P(yes)]]

Citation

@misc{assertion_sentence_has_time_reference,
  author = {Stanford SCALE Initiative},
  title  = {Assertion classifier: sentence has time reference},
  year   = {2026},
  url    = {https://huggingface.co/StanfordSCALE/assertion_sentence_has_time_reference}
}