Implementation of Memory Network in pyTorch
This repo including implementations of End-to-End Memory Network and Key-Value Memory Network
Dataset used here is WikiMovie. Note that the preprocessing scripts are borrowed from original repo
Code (mainly in pytorch)
- config.py: some global configurations for data dir and training detail
- gen_dict.py: generate dict
- data_loader.py: load preprocessed data
- model.py: model for training similarities and readers
- train_*.py: training script for experimental model
- plot.py: plot the graph for report
Run
- run setup_processed_data.sh to get the processed data
- run gen_dict.py to generate the dictionary file
- run gen_sim_data.py to generate the data for training similarity function
- run the train_mlp.py to train the similarity function
- run gen_reader_data.py to generate the data for reader
- run train_lstm.py, train_memory_network.py, train_kv_mm.py to train the corresponding model