Module Import alias identifier ImportFrom alias identifier ImportFrom alias Import alias ClassDef bases NameLoad body FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam NameParam NameParam defaults Num AttributeLoad NameLoad attr Num NameLoad NameLoad body Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr NameLoad NameLoad NameLoad Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam NameParam NameParam NameParam NameParam NameParam NameParam defaults Num Num Num Num body Expr Call AttributeLoad NameLoad attr Str Expr Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr Expr Call AttributeLoad NameLoad attr Str Expr Call AttributeLoad NameLoad attr BinOpMod Str TupleLoad Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Call NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Expr Call AttributeLoad NameLoad attr BinOpMod Str TupleLoad Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Call NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Expr Call AttributeLoad NameLoad attr BinOpMod Str TupleLoad AttributeLoad NameLoad attr Call NameLoad AttributeLoad NameLoad attr Expr Call AttributeLoad NameLoad attr Str Assign NameStore NameLoad Assign NameStore Num While BoolOpAnd CompareLt NameLoad NameLoad UnaryOpNot NameLoad body AugAssignAdd NameStore Num If CompareEq BinOpMod NameLoad AttributeLoad NameLoad attr Num body AugAssignMult SubscriptStore SubscriptLoad NameLoad Index Str Index Str AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr Assign NameStore ListLoad For NameStore Call NameLoad NameLoad body Assign NameStore Call NameLoad NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Expr Call AttributeLoad NameLoad attr NameLoad Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad keyword Num Expr Call AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore BinOpSub Call AttributeLoad NameLoad attr NameLoad If CompareEq BinOpMod NameLoad AttributeLoad NameLoad attr Num body If CompareEq NameLoad Num body Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad Call NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr orelse Assign NameStore ListLoad For NameStore Call NameLoad NameLoad body Assign NameStore Call NameLoad NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Expr Call AttributeLoad NameLoad attr NameLoad Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad keyword Num If CompareIsNot NameLoad NameLoad body If CompareEq NameLoad Num body Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad Call NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr orelse Assign NameStore ListLoad For NameStore Call NameLoad NameLoad body Assign NameStore Call NameLoad NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Expr Call AttributeLoad NameLoad attr NameLoad Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad keyword Num orelse Assign SubscriptStore AttributeLoad NameLoad attr Index NameLoad BinOpMult ListLoad Num Call NameLoad Call AttributeLoad SubscriptLoad NameLoad Index Str attr Assign NameStore BinOpMod Str TupleLoad Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr NameLoad NameLoad AugAssignAdd NameStore BinOpAdd BinOpAdd BinOpAdd BinOpAdd Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Str Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Str Call AttributeLoad Str attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Assign NameStore ListComp Call NameLoad NameLoad comprehension NameStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad AugAssignAdd NameStore ListComp Call NameLoad NameLoad comprehension NameStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad AugAssignAdd NameStore ListComp Call NameLoad NameLoad comprehension NameStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad AugAssignMod NameStore Call NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad If BoolOpAnd CompareIsNot AttributeLoad NameLoad attr NameLoad CompareEq BinOpMod NameLoad AttributeLoad NameLoad attr Num body If CompareIsNot AttributeLoad NameLoad attr NameLoad body Expr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad AttributeLoad AttributeLoad NameLoad attr attr Expr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Str Expr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Str Expr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Call AttributeLoad SubscriptLoad NameLoad Index Str attr Str Expr Call AttributeLoad NameLoad attr Expr Call AttributeLoad AttributeLoad NameLoad attr attr If CompareIsNot AttributeLoad NameLoad attr NameLoad body Expr Call AttributeLoad AttributeLoad NameLoad attr attr decorator_list decorator_list
N/A N/A numpy np utils env_paths paths base Train N/A time TrainModel N/A Train N/A __init__ N/A N/A self model anneal_lr anneal_lr_freq output_freq pickle_f_custom_freq f_custom_eval N/A 1.0 N/A np inf 1 None None N/A N/A N/A N/A N/A super TrainModel self __init__ model pickle_f_custom_freq f_custom_eval N/A N/A self anneal_lr anneal_lr N/A N/A self output_freq output_freq N/A N/A self anneal_lr_freq anneal_lr_freq N/A train_model N/A N/A self f_train train_args f_test test_args f_validate validation_args n_train_batches n_valid_batches n_test_batches n_epochs N/A 600 1 1 100 N/A N/A N/A N/A self write_to_logger ### MODEL PARAMS ### N/A N/A N/A self write_to_logger N/A N/A N/A self model model_info N/A N/A N/A self write_to_logger ### TRAINING PARAMS ### N/A N/A N/A self write_to_logger N/A Train -> %s: %s N/A N/A N/A ; join N/A N/A N/A train_args N/A inputs keys N/A str N/A N/A N/A train_args N/A inputs values N/A N/A N/A self write_to_logger N/A Test -> %s: %s N/A N/A N/A ; join N/A N/A N/A test_args N/A inputs keys N/A str N/A N/A N/A test_args N/A inputs values N/A N/A N/A self write_to_logger N/A Anneal LR %0.4f after %i. N/A N/A self anneal_lr N/A int N/A self anneal_lr_freq N/A N/A N/A self write_to_logger ### TRAINING MODEL ### N/A done_looping False N/A epoch 0 N/A N/A N/A epoch n_epochs N/A done_looping N/A N/A epoch 1 N/A N/A N/A epoch N/A self anneal_lr_freq 0 N/A N/A N/A N/A train_args N/A inputs N/A learningrate N/A self anneal_lr N/A start_time N/A N/A time time N/A train_outputs N/A N/A i N/A xrange n_train_batches N/A N/A train_output N/A f_train i N/A N/A N/A train_args N/A inputs values N/A N/A N/A train_outputs append train_output N/A N/A N/A self eval_train N/A epoch N/A N/A np mean N/A N/A np array train_outputs axis 0 N/A N/A N/A N/A self model after_epoch N/A end_time N/A N/A N/A time time start_time N/A N/A N/A epoch N/A self output_freq 0 N/A N/A N/A n_test_batches 1 N/A N/A N/A N/A self eval_test N/A epoch N/A f_test N/A N/A N/A test_args N/A inputs values N/A N/A test_outputs N/A N/A i N/A xrange n_test_batches N/A N/A test_output N/A f_test i N/A N/A N/A test_args N/A inputs values N/A N/A N/A test_outputs append test_output N/A N/A N/A self eval_test N/A epoch N/A N/A np mean N/A N/A np array test_outputs axis 0 N/A N/A f_validate None N/A N/A N/A n_valid_batches 1 N/A N/A N/A N/A self eval_validation N/A epoch N/A f_validate N/A N/A N/A validation_args N/A inputs values N/A N/A valid_outputs N/A N/A i N/A xrange n_valid_batches N/A N/A valid_output N/A f_validate i N/A N/A N/A validation_args N/A inputs values N/A N/A N/A valid_outputs append valid_output N/A N/A N/A self eval_validation N/A epoch N/A N/A np mean N/A N/A np array valid_outputs axis 0 N/A N/A N/A N/A self eval_validation N/A epoch N/A N/A 0.0 N/A len N/A N/A N/A validation_args N/A outputs keys N/A output_str N/A [epoch,time,train[%s],test[%s],valid[%s]];%i;%0.2f; N/A N/A N/A , join N/A N/A N/A train_args N/A outputs keys N/A N/A , join N/A N/A N/A test_args N/A outputs keys N/A N/A , join N/A N/A N/A validation_args N/A outputs keys epoch end_time N/A output_str N/A N/A N/A N/A N/A N/A ; join N/A N/A N/A train_args N/A outputs values ; N/A N/A ; join N/A N/A N/A test_args N/A outputs values ; N/A N/A ; join N/A N/A N/A validation_args N/A outputs values N/A outputs N/A N/A float o N/A o N/A N/A self eval_train N/A epoch N/A outputs N/A N/A float o N/A o N/A N/A self eval_test N/A epoch N/A outputs N/A N/A float o N/A o N/A N/A self eval_validation N/A epoch N/A output_str N/A tuple outputs N/A N/A N/A self write_to_logger output_str N/A N/A N/A N/A self pickle_f_custom_freq None N/A N/A epoch N/A self pickle_f_custom_freq 0 N/A N/A N/A N/A self custom_eval_func None N/A N/A N/A N/A self custom_eval_func N/A self model N/A N/A paths get_custom_eval_path epoch N/A N/A self model root_path N/A N/A N/A self plot_eval N/A self eval_train N/A N/A N/A train_args N/A outputs keys _train N/A N/A N/A self plot_eval N/A self eval_test N/A N/A N/A test_args N/A outputs keys _test N/A N/A N/A self plot_eval N/A self eval_validation N/A N/A N/A validation_args N/A outputs keys _validation N/A N/A N/A self dump_dicts N/A N/A N/A N/A self model dump_model N/A N/A N/A self pickle_f_custom_freq None N/A N/A N/A N/A N/A self model dump_model N/A N/A