agaresd commited on
Commit
e18dfad
·
verified ·
1 Parent(s): 5686e98

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -31,13 +31,11 @@ While the model successfully predicts dialogue acts, it faces challenges in emot
31
  - **Emotion Classifier**: Predicts one of the annotated emotions (e.g., happiness, sadness, anger).
32
 
33
  ### Input
34
- - **input_ids**: Tokenized input sequences of dialogues.
35
- - **attention_mask**: Binary mask indicating valid tokens in the input sequence.
36
 
37
  ### Output
38
  - **act_output**: Predicted dialogue act class.
39
  - **emotion_output**: Predicted emotion class.
40
- - **kl_div**: KL divergence loss for SAE regularization.
41
 
42
  ---
43
 
 
31
  - **Emotion Classifier**: Predicts one of the annotated emotions (e.g., happiness, sadness, anger).
32
 
33
  ### Input
34
+ - **sentence**: Single word or sentence
 
35
 
36
  ### Output
37
  - **act_output**: Predicted dialogue act class.
38
  - **emotion_output**: Predicted emotion class.
 
39
 
40
  ---
41