Beispiel #1
0
 def input_data_fn():
     dataset = winequality.get_dataset(WINE_EQUALITY_FILE,
                                       split="train")
     return (dataset.map(convert_to_tensor).shuffle(1000).batch(
         128).repeat())
Beispiel #2
0
 def train_input_fn():
     dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="train")
     return (dataset.shuffle(1000).batch(128).repeat())
Beispiel #3
0
 def eval_input_fn():
     dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="test")
     return (dataset.shuffle(1000).batch(128))
Beispiel #4
0
 def validation_data_fn():
     dataset = winequality.get_dataset(WINE_EQUALITY_FILE, split="test")
     return (dataset.map(convert_to_tensor).shuffle(1000).batch(128))