def test_collect_activations(csv_filename):
model = None
try:
# This will reset the layer numbering scheme TensorFlow uses.
# Otherwise, when we load the model, its layer names will be appended
# with "_1".
tf.keras.backend.reset_uids()
model = _train(*_prepare_data(csv_filename))
model_path = os.path.join(model.exp_dir_name, 'model')
layers = _get_layers(model_path)
assert len(layers) > 0
tf.keras.backend.reset_uids()
with tempfile.TemporaryDirectory() as output_directory:
filenames = collect_activations(model_path,
layers,
data_csv=csv_filename,
output_directory=output_directory)
assert len(filenames) > len(layers)
finally:
if model and model.exp_dir_name:
shutil.rmtree(model.exp_dir_name, ignore_errors=True)
def _collect_activations(model_path, layers, csv_filename, output_directory):
return collect_activations(model_path,
layers,
data_csv=csv_filename,
output_directory=output_directory)