Skip to content
This repository has been archived by the owner on May 6, 2023. It is now read-only.

Releases: AFAgarap/pt-datasets

v0.1.3

04 Nov 08:38
Compare
Choose a tag to compare

Install this package through PyPi,

$ pip install pt-datasets

To enable GPU support for t-SNE, run the installation script for tsnecuda,

$ git clone https://github.com/AFAgarap/pt-datasets.git
$ cd pt-datasets
$ bash setup/install_tsnecuda

Initial release

02 Nov 17:24
Compare
Choose a tag to compare

This repository is meant for easier and faster access to the following image classification datasets: MNIST, Fashion-MNIST, EMNIST-Balanced, CIFAR10, and SVHN. Using this repository, one can load the aforementioned datasets in a ready-to-use fashion for PyTorch models. Additionally, this can be used to load the low-dimensional features of the aforementioned datasets, encoded using PCA, t-SNE, or UMAP.

Features

  • Load MNIST, Fashion-MNIST, EMNIST-Balanced, CIFAR10, and SVHN PyTorch datasets.
  • Create a data loader for a dataset, ready for model use.
  • Encode features to lower-dimensional space using PCA, t-SNE, and UMAP.