datagnosis.plugins.core.datahandler module

class DataHandler(X: Union[pandas.core.frame.DataFrame, numpy.ndarray, torch.Tensor], y: Union[pandas.core.series.Series, numpy.ndarray, torch.Tensor], batch_size: Optional[int] = None, reproducible: bool = True, **kwargs: Any)

Bases: object

toJson() → str

Returns a JSON representation of the DataHandler object.

Returns

A JSON representation of the DataHandler object in the form of a string.

Return type

str

class IndexedDataset(*tensors: torch.Tensor)

Bases: torch.utils.data.dataset.TensorDataset

tensors: Tuple[torch.Tensor, ...]