tardis.data

Provides samples of more complicated data sets.

tardis.data.sample_vision_data_set(name, n_samples)

Sample vision data set.

Parameters:

• name (str) – Name of the data set. Currently, only “MNIST” and “FashionMNIST” are supported here.

• n_samples (int) – Number of samples to retrieve.

Returns:

Sampled data points

Return type:

np.array