├── .gitignore ├── LICENSE ├── README.md ├── figs ├── correct-rate.png ├── dnd-lstm-cell.png ├── eg-0.png ├── eg-1.png └── pc-v.png ├── requirements.txt └── src ├── contextual-choice.ipynb ├── contextual-choice.py ├── model ├── A2C.py ├── DND.py ├── DNDLSTM.py ├── __init__.py └── utils.py ├── task ├── ContextualChoice.py └── __init__.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/README.md -------------------------------------------------------------------------------- /figs/correct-rate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/figs/correct-rate.png -------------------------------------------------------------------------------- /figs/dnd-lstm-cell.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/figs/dnd-lstm-cell.png -------------------------------------------------------------------------------- /figs/eg-0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/figs/eg-0.png -------------------------------------------------------------------------------- /figs/eg-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/figs/eg-1.png -------------------------------------------------------------------------------- /figs/pc-v.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/figs/pc-v.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | matplotlib 4 | seaborn 5 | -------------------------------------------------------------------------------- /src/contextual-choice.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/contextual-choice.ipynb -------------------------------------------------------------------------------- /src/contextual-choice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/contextual-choice.py -------------------------------------------------------------------------------- /src/model/A2C.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/model/A2C.py -------------------------------------------------------------------------------- /src/model/DND.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/model/DND.py -------------------------------------------------------------------------------- /src/model/DNDLSTM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/model/DNDLSTM.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/model/utils.py -------------------------------------------------------------------------------- /src/task/ContextualChoice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/task/ContextualChoice.py -------------------------------------------------------------------------------- /src/task/__init__.py: -------------------------------------------------------------------------------- 1 | from .ContextualChoice import * 2 | -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qihongl/dnd-lstm/HEAD/src/utils.py --------------------------------------------------------------------------------