A neural net project for Eulerian path in Graphs
Run the Eulerian task from the euleraian_path_task folder
command to run
python run_tasks.py --experiment_name="Eulerian_path" --mann=ntm --task="graph"
Run the Nback Image task from the nback_images folder
command to run
python run_tasks.py --experiment_name="N-BackImages" --mann=ntm --task="nback"
Arguments that can be passed to change the hyper parameters are-
--mann', default='ntm', help='none | ntm'
--num_layers', default=1
--num_units', default=100
--num_memory_locations', default=128
--memory_size', default=20
--num_read_heads', default=1
--num_write_heads', default=1
--conv_shift_range', default=1, help='only necessary for ntm'
--clip_value', default=20, help='Maximum absolute value of controller and outputs.'
--init_mode', default='constant', help='learned | constant | random'
--optimizer', default='Adam', help='RMSProp | Adam'
--learning_rate', default=0.01
--max_grad_norm', default=50
--num_train_steps', default=402
--batch_size', default=32
--eval_batch_size', default=640
--curriculum', default='none',
help='none | uniform | naive | look_back | look_back_and_forward | prediction_gain'
--pad_to_max_seq_len', default=False
--task', default='nback', help='graph' | nback'
--num_bits_per_vector', default=784
--max_seq_len', default=5
--store_result', default=True, help='stores result'
--verbose', default=False, help='if true prints lots of feedback'
--experiment_name', required=True
--job-dir', required=False
--steps_per_eval', default=200
--use_local_impl', default=True,
help='whether to use the repos local NTM implementation or the TF contrib version'