s5tf-team / datasets

Datasets ready to use with Swift for TensorFlow

Datasets

Datasets is a collection of ready-to-use S5TFDataLoaders.

Documentation

Datasets can be loaded in just a single line of code:

Iris.train

To loop over batches, use batched():

for batch in Iris.train.batched(32) {
  print(batch.data, batch.labels)
}

View an interactive Google Colab example here.

Contributing ❤️

Thanks for even considering contributing.

Make sure to run swiftlint on your code. If you are not sure about how to format something, refer to the Google Swift Style Guide.

Please link to the completed GitHub Actions build test in your fork with your PR.