loaders.load_MNIST_with_embeddings

pai4sk.simsearch.loaders.load_MNIST_with_embeddings(image_file_path, label_file_path, norm='l1', use_sparse=True)

Loads MNIST dataset into arrays W, X, labels, ids

Parameters:
  • image_file_path (string) – file path location for the MNIST images
  • label_file_path (string) – file path location for the MNIST labels
  • norm (string (default='l1')) – l2 if the user wants to normalize the data using l2 normalization
  • use_sparse (bool (default='True')) – False if the user wants to load MNIST in non-sparse format
Returns:

  • W (array_like, shape (n_features, n_dimensions)) – Embedding vectors
  • X (array_like, sparse_matrix, shape (n_samples, n_features)) – Feature vectors
  • labels (array-like, shape (n_samples,)) – labels are the class labels of the samples in X
  • ids (array_like, shape (n_samples,)) – ids are the ids of the samples in X