tensorflow / swift

Swift for TensorFlow
https://tensorflow.org/swift
Apache License 2.0
6.12k stars 608 forks source link

Update the model training walkthrough tutorial to use Epochs instead of TensorFlow.Datasets #466

Closed BradLarson closed 4 years ago

BradLarson commented 4 years ago

The model training walkthrough tutorial notebook uses the Iris CSV-based dataset in its example code. The implementation of this relies on the old TensorFlow.Dataset API, which has been deprecated in favor of the new Epochs API.

This notebook (and supporting code file) should be updated to use the new Epochs API to parse the Iris dataset and perform training.