deepSIP.dataset

class deepSIP.dataset.NumpyXDataset(X, device='auto')

Bases: torch.utils.data.dataset.Dataset

create torch tensor dataset from X numpy ndarray

Parameters:
X : np.ndarray of shape (number of spectra, number of wavelength bins)

pre-processed spectra

device : str, optional

device to push tensors to, defaults to GPU if available

Attributes:
X : torch.tensor of shape (number of spectra, 1, number of wavelength bins)
class deepSIP.dataset.NumpyXYDataset(X, Y, **kwargs)

Bases: deepSIP.dataset.NumpyXDataset

create torch tensor dataset from X, Y numpy ndarrays

Parameters:
X : np.ndarray of shape (number of spectra, number of wavelength bins)

pre-processed spectra

Y : np.ndarray of shape (number of spectra, number of targets)

targets

**kwargs

arbitrary keyword arguments for NumpyXDataset constructor

Attributes:
X : torch.tensor of shape (number of spectra, 1, number of wavelength bins)
Y : torch.tensor of shape (number of spectra, number of targets)