Skip to content

Instantly share code, notes, and snippets.

View Ewen2015's full-sized avatar
🥥

Ewen Wang Ewen2015

🥥
View GitHub Profile
@Ewen2015
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.