zizhaozhang
/ tf-experiment-template.py
Created
April 7, 2019 07:11
— forked from damienpontifex/tf-experiment-template.py
A template for a custom tensorflow estimator and experiment with python3 typings for desired parameter types
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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. |