UCIRegression¶
- class torch_uncertainty.datasets.regression.UCIRegression(root, transform=None, target_transform=None, dataset_name='energy', download=False, seed=42, shuffle=True)[source]¶
The UCI regression datasets.
- Parameters:
root (str) – Root directory of the datasets.
train (bool, optional) – If True, creates dataset from training set, otherwise creates from test set.
transform (callable, optional) – A function/transform that takes in a numpy array and returns a transformed version.
target_transform (callable, optional) – A function/transform that takes in the target and transforms it.
dataset_name (str, optional) – The name of the dataset. One of “boston-housing”, “concrete”, “energy”, “kin8nm”, “naval-propulsion-plant”, “power-plant”, “protein”, “wine-quality-red”, and “yacht”.
download (bool, optional) – If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.
- Note - Ethics:
You may want to avoid using the boston-housing dataset because of ethical concerns.
- Note - License:
The licenses of the datasets may differ from TorchUncertainty’s license. Check before use.