🥥

Ewen Wang Ewen2015

🥥

Nothing can dim the light that shines from within.

Ewen2015 / tf-experiment-template.py

Created December 12, 2018 14:06 — forked from damienpontifex/tf-experiment-template.py

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.