google-research-datasets/go_emotions
Viewer β’ Updated β’ 265k β’ 28.1k β’ 284
How to use Ayaannnnnnn/midnight-wash-counselor-e5 with Transformers.js:
// npm i @huggingface/transformers
import { pipeline } from '@huggingface/transformers';
// Allocate pipeline
const pipe = await pipeline('text-classification', 'Ayaannnnnnn/midnight-wash-counselor-e5');A small classifier used by the confession room in Midnight Wash, a 3D late-night laundromat web app. It sorts a short worry (Korean or English) into one of 11 emotions, and the app replies with reviewed, hand-written sentences for that emotion. The model never writes replies itself.
onnx/model_quantized.onnx, about 118 MB) for transformers.js; runs on CPU in the browsertired, hurt, guilt, anxious, lonely, grief, angry, future, burnout, crisis, otherquery: (as in training)import { pipeline } from '@huggingface/transformers';
const clf = await pipeline('text-classification', 'Ayaannnnnnn/midnight-wash-counselor-e5', { dtype: 'q8' });
const [top] = await clf('query: μ€λ νμ¬μμ ν루 μ’
μΌ νΌλμ λ무 μ§μ³€μ΄μ.');
KOTE-MIT.txt)GoEmotions-Apache-2.0.txt)Public labels were mapped onto the 11 emotions; lonely, future and crisis come only from the synthetic set.
| Set | Model alone | In the app (crisis rules first, keywords when the model scores below 0.4) |
|---|---|---|
| 28 held-out worry sentences | 20/28 | 25/28 |
| KOTE test split, mapped (86 Korean comments) | 58/86 | 58/86 |
| GoEmotions test split, mapped (60 English comments) | 37/60 | 36/60 |
The public test sets are comments about news, products and posts rather than personal worries, and their labels only partly match ours, so use them to compare versions rather than as absolute accuracy.
crisis label. In the 28-sentence test the model labeled crisis phrases as future. The app checks crisis phrases with rules first and always shows a crisis line (in Korea, 109) when they match.Base model
intfloat/multilingual-e5-small