Skip to content

NYCU-MLLab/Contrastive-Disentanglement-Learning-for-Empathetic-Dialogue-Generation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Contrastive-Disentanglement-Learning-for-Empathetic-Dialogue-Generation

Installation

pip install -r requirements.txt

Training

go to the directory source then do

python3 train_multi.py

Model weight

The pre-trained is provide in Weight. Make sure the weight show be put in the right direction. Please create a directoryMy_model_pth inside the directory Empathetic_Dialogue or the directory NYCUKA. Then you can run the evaluation.

Evaluation

go "outside" the directory source then do

python3 eval_multi_continuous.py

Sampleing

Make sure to run the evluation first. You can create the directory what you want to put the output file inside and then you can run the file example.ipynb

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 84.5%
  • Jupyter Notebook 15.5%