CODEWITHSUNDEEP
pythonnumpyh5py
memecs
memecs

Reputation: 7564

h5py: slicing dataset without loading into memory

Is it possible to slice an h5py dataset in two subsets without actually loading them into memory? E.g.:

dset = h5py.File("/2tbhd/tst.h5py","r")

X_train = dset['X'][:N/2]
X_test  = dset['X'][N/2:-1]

Upvotes: 5

Views: 2069

Answers (1)

Stuart Berg
Stuart Berg

Reputation: 18141

No.

You would need to implement your own class to act as a view on the dataset. An old thread on the h5py mailing list indicates that such a DatasetView class is theoretically possible to implement using HDF5 dataspaces, but probably not worth it for many use cases. Element-wise access would be very slow compared to a normal numpy array (assuming you can fit your data into memory).

Edit: If you want to avoid messing with HDF5 data spaces (whatever that means), you might settle for a simpler approach. Try this gist I just wrote. Use it like this:

dset = h5py.File("/2tbhd/tst.h5py","r")

from simpleview import SimpleView
X_view = SimpleView(dset['X'])

# Stores slices, but doesn't load into memory
X_train = X_view[:N/2]
X_test  = X_view[N/2:-1]

# These statements will load the data into memory.
print numpy.sum(X_train)
print numpy.array(X_test)[0]

Note that the slicing support in this simple example is somewhat limited. If you want full slicing and element-wise access, you'll have to copy it into a real array:

X_train_copy = numpy.array(X_train)

Upvotes: 4

Related Questions