Alice Cecile alice-i-cecile

Mad scientist, ecologist, game (engine) developer. Building Bevy.

damienpontifex / tf-experiment-template.py

Last active December 11, 2025 01:00

A template for a custom tensorflow estimator and experiment with python3 typings for desired parameter types

	import argparse
	import psutil
	import tensorflow as tf
	from typing import Dict, Any, Callable, Tuple

	## Data Input Function
	def data_input_fn(data_param,
	batch_size:int=None,
	shuffle=False) -> Callable[[], Tuple]:
	"""Return the input function to get the test data.