swift-datascience icon indicating copy to clipboard operation
swift-datascience copied to clipboard

swift-datascience

Project Structure

  • MNIST.swift : Code to load the MNIST data and convert it into batch of tensors
  • Swift_Data_Science_MNIST.ipynb : Notebook of the project, can be directly run in Colab
  • main_model.swift : Main file of the model
  • t10k-images-idx3-ubyte, t10k-labels-idx1-ubyte, train-images-idx3-ubyte, train-labels-idx1-ubyte : MNIST train and test files

License

All the code in this repository is under the MIT license.

Credits to

I would like to thank these open source projects: