edflow.iterators.tf_iterator module

Summary

Classes:

TFHookedModelIterator

Reference

class edflow.iterators.tf_iterator.TFHookedModelIterator(config, root, model, datasets, hook_freq=100, num_epochs=100, hooks=[], bar_position=0, nogpu=False, desc='')

Bases: edflow.iterators.model_iterator.PyHookedModelIterator

make_feeds(batch)

run(fetches, feed_dict)

Runs all fetch ops and stores the results.

Parameters

• fetches (dict) – name: Callable pairs.

• feed_dict (dict) – Passed as kwargs to all fetch ops

Returns

name: results pairs.

Return type

dict

iterate(batch_iterator, validation_batch_iterator=None)

Iterates over the data supplied and feeds it to the model.

Parameters

• batch_iterator (Iterable) – Iterable returning training data.

• batch_iterator_validation (Iterable) – Iterable returning validation data or None

property session