def input_data_fn(): dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="train") return (dataset.map(convert_to_tensor).shuffle(1000).batch( 128).repeat())
def train_input_fn(): dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="train") return (dataset.shuffle(1000).batch(128).repeat())
def eval_input_fn(): dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="test") return (dataset.shuffle(1000).batch(128))
def validation_data_fn(): dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="test") return (dataset.map(convert_to_tensor).shuffle(1000).batch(128))