Module: tfl.test_utils

Helpers to train simple model for tests and print debug output.


class TimeTracker: Tracks time.


assert_sonnet_equivalent_to_keras(...): Asserts that a Sonnet module is "equivalent" to a Keras layer.

get_hypercube_interpolation_fn(...): Returns function which does hypercube interpolation.

get_linear_lattice_interpolation_fn(...): Returns function which does lattice interpolation.

run_training_loop(...): Trains models and prints debug info.

sample_uniformly(...): Deterministically generates num_point random points within bounds.

two_dim_mesh_grid(...): Generates uniform 2-d mesh grid for 3-d surfaces visualisation via pyplot.