edflow.iterators.tf_evaluator module

Summary

Classes:

TFBaseEvaluator

Reference

class edflow.iterators.tf_evaluator.TFBaseEvaluator(*args, desc='Eval', hook_freq=1, num_epochs=1, **kwargs)

Bases: edflow.iterators.tf_iterator.TFHookedModelIterator

__init__(*args, desc='Eval', hook_freq=1, num_epochs=1, **kwargs)

New Base evaluator restores given checkpoint path if provided, else scans checkpoint directory for latest checkpoint and uses that

Parameters

• desc (str) – a description for the evaluator. This description will be used during the logging.

• hook_freq (int) – Frequency at which hooks are evaluated.

• num_epochs (int) – Number of times to iterate over the data.

initialize(checkpoint_path=None)

define_graph()

step_ops()

Defines ops that are called at each step.

Returns

Return type

The operation run at each step.