https://github.com/qkaren/COLD_decoding

https://github.com/qkaren/COLD_decoding

Last synced: 3 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/qkaren/COLD_decoding
• Owner: qkaren
• License: apache-2.0
• Created: 2022-02-23T18:30:50.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-06-09T22:01:43.000Z (over 2 years ago)
• Last Synced: 2024-07-03T23:45:15.890Z (4 months ago)
• Language: Python
• Size: 674 KB
• Stars: 96
• Watchers: 5
• Forks: 22
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Code for "COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamics"

This is the code for the following paper: 

[COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamics] (https://arxiv.org/pdf/2202.11705.pdf) \

Lianhui Qin, Sean Welleck, Daniel Khashabi, Yejin Choi 

**1) Setup Environment**

```

pip install -r requirements.txt

```

**2) Download this Github**

```

git clone https://github.com/qkaren/COLD_decoding.git

```

**3) Run Command for COLD Decoding**

* CommonGen

```

sh commongen.sh

```

* Abductive Reasoning 

```

sh abductive.sh

```

* Counterfactual Reasoning

```

sh counterfactual.sh

```

**4) Rank the generation**

will add soon