Load and save csr_matrix dataset as numpy npz file

load_save_npz_sparse_dataset.py

import numpy as np
from scipy.sparse import csr_matrix


def save_dataset(filename, x, y):
    np.savez(filename, data=x.data, indices=x.indices,
             indptr=x.indptr, shape=x.shape, target=y)


def load_dataset(filename):
    loader = np.load(filename)
    return csr_matrix((loader['data'], loader['indices'], loader['indptr']),
                      shape=loader['shape']), loader['target']