Skip to content

jamesHargreaves12/incremental_beam_manipulation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Comands to run to get experiments working:

The comands setup the data from the e2e-dataset for use in this thesis

  • cd tgen/e2e-challenge/input
  • ./convert.py -a name,near -n ../../../e2e-dataset/trainset.csv train
  • ./convert.py -a name,near -n -m ../../../e2e-dataset/devset.csv devel
  • cd ../../..

To train the Seq2seq model:

  • python3 train_seq2seq.py -c new_configs/model_configs/seq2seq_model.yaml

To test the vanilla beam search:

  • python3 get_results.py -c new_configs/vanilla_beam_search.yaml

To train beam manipulator model:

  • python3 get_results.py -c new_configs/setup_bm_traindata.yaml
  • python3 train_beam_manipulator.py -c new_configs/model_configs/bm_model.yaml

To test beam manipulator:

  • python3 get_results.py -c new_configs/bm_test.yaml

To view the bleu scores of the experiments:

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages