Skip to content

Instantly share code, notes, and snippets.

@hanneshapke

hanneshapke/using_tft_with_model_fit.py

Created March 9, 2020 18:04
Show Gist options
  • Save hanneshapke/92729a00ecb72d12fcd32a8bbc4edd0b to your computer and use it in GitHub Desktop.
Save hanneshapke/92729a00ecb72d12fcd32a8bbc4edd0b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
using_tft_with_model_fit.py
tf_transform_output = tft.TFTransformOutput(fn_args.transform_output)
train_dataset = _input_fn(fn_args.train_files, tf_transform_output, 32)
eval_dataset = _input_fn(fn_args.eval_files, tf_transform_output, 32)
...
model.fit(
train_dataset,
validation_data=eval_dataset,
...
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment