Skip to content

a CMA-ES based hyperparameter optimization tool for NMT.

License

Notifications You must be signed in to change notification settings

marvinzh/cma_es

Repository files navigation

CMA-ES

a CMA-ES based hyperparameter optimization tool.

Prerequisite

pip install cma

basic usage:

./run.sh 0

Usage

evo_single.py : load initial hyperparameter or previous checkpoint, generate genes for current generation.

gene2conf.py: load genes for current generation, convert them to model description file.

reporter.py: load the evalution result file (e.g. BLEU score) output by each model. report them in a file.

toy_nmt.py: Script to generate random score as a fake evaluation score for testing. (format from sockeye-recipe)

Copyright

Copyright (c) 2018 Shinozaki-lab, Tokyo Institute of Technology.

About

a CMA-ES based hyperparameter optimization tool for NMT.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published