Generative-MultiEmo

KO | EN

gitlite — search

#python #react #llm #golang #android #machine-learning #deep-learning #nodejs #windows #vue #awesome-list #pytorch #openai #mcp

Generative-MultiEmo

(★ 23)

Source code for NLPCC 2022 paper: Prompt-Based Generative Multi-label Emotion Prediction with Label Contrastive Learning

Generative-MultiEmo Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Prompt-Based Generative Multi-label Emotion Prediction with Label Contrastive Learning
This is the source code for NLPCC 2022 paper: [Prompt-Based Generative Multi-label Emotion Prediction with Label Contrastive Learning](https://link.springer.com/chapter/10.1007/978-3-031-17120-8_43)
## 1. Environments

```
- python (3.9.5)
- cuda (11.4)
```

## 2. Dependencies

```
- torch (1.10.0)
- pytorch-lightning (1.5.1)
- transformers (4.12.3)
- pandas (1.3.4)
- scikit-learn (1.0.1)
- tensorboard (2.7.0)
- ekphrasis (0.5.1)
```

## 3. Dataset

- [GoEmotions](https://github.com/google-research/google-research/tree/master/goemotions)
- [SemEval 2018 E-c](https://competitions.codalab.org/competitions/17751#learn_the_details)

For SemEval 2018 dataset, we utilized a tweet pre-process tool named [ekphrasis](https://github.com/cbaziotis/ekphrasis) to convert words to lower case, normalise user mentions, urls and repeated-characters. We provided the processed data additionally.

## 4. Training

For GoEmotions dataset, please run:

```bash
>> python main.py --CLP
```

For SemEval 2018 dataset, please run:

```bash
>> python main.py --dataset 'SemEvalEc' --num_beams 2 --CLP
```

For ablation study, please run:

```bash
>> python main.py
>> python main.py --dataset 'SemEvalEc' --num_beams 2
```

## 5. Ciation

If you find our work useful for your application or reaserch, please kindly cite our paper:

```
@InProceedings{Generative-MultiEmo,
author={Chai, Yuyang
and Teng, Chong
and Fei, Hao
and Wu, Shengqiong
and Li, Jingye
and Cheng, Ming
and Ji, Donghong
and Li, Fei},
title={Prompt-Based Generative Multi-label Emotion Prediction with Label Contrastive Learning},
booktitle={Natural Language Processing and Chinese Computing},
year={2022},
pages={551--563},
}
```

Generative-MultiEmo

Generative-MultiEmo Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Generative-MultiEmo

Generative-MultiEmo Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README