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)