load_snap_file

pai4sk.sml_io.load_snap_file(filename)

Data loading from Snap formatted data file. It supports both local and distributed(MPI) method of loading data. For MPI execution this can be used for distributed SnapML training and inference.

Parameters

filename (str) – The file where the data resides in snap format.

Returns

  • X (scipy.sparse matrix or ndarray of shape (n_samples, n_features))

  • y (ndarray of shape (n_samples,))