chrlukas/

stories-emotion-c8

This model is intended to predict emotions (valence, arousal) in written stories. For all details see the paper and the accompanying github repo.

As described in the paper, this model is finetuned from DeBERTaV3-large and predicts sentence-wise valence/arousal values between 0 and 1.

This particular checkpoint was trained with a window size of 8.

All available checkpoints and their performance measured by Concordance Correlation Coefficient (CCC):

We provide the best out of 5 seeds for each context size. Hence, the numbers in this table differ from the result table in the paper, where the mean performance across 5 seeds is reported.

Technically, this model predicts token-wise valence/arousal values. Sentences are concatenated via the [SEP] token, where the valence/arousal predictions for an [SEP] token are meant to be the predictions for the sentence preceding it. All other tokens' predictions should be ignored. For reference, see the figure in the paper:

The accompanying repo provides a convenient script to use the model for prediction.

This model is intended to predict emotions (valence, arousal) in written stories. It was mainly trained on stories for children. Please note that the model is not production-ready and provided here for demonstration purposes only. For details on the datasets used, please refer to the paper.

In the github repository, a convenient script to predict V/A in existing texts is provided. Example call:

Please see the Limitations section in the paper. Please note that the model is not production-ready and provided here for demonstration purposes only.

For further inquiries, please contact lukas1[dot]christ[at]uni-a[dot].de