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:
image
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: