yamle.data package#
- yamle.data.data_factory(data_type)[source]#
This function is used to create a data module instance based on the data type.
- Return type:
Type[BaseDataModule]
Submodules#
- yamle.data.augmentations module
- yamle.data.classification module
TorchvisionClassificationDataModuleTinyImageNetClassificationDataModuleTinyImageNetClassificationDataModule.inputs_dimTinyImageNetClassificationDataModule.outputs_dimTinyImageNetClassificationDataModule.targets_dimTinyImageNetClassificationDataModule.meanTinyImageNetClassificationDataModule.stdTinyImageNetClassificationDataModule.available_transformsTinyImageNetClassificationDataModule.available_test_augmentationsTinyImageNetClassificationDataModule.test_augmentations
TorchvisionClassificationDataModuleMNISTTorchvisionClassificationDataModuleMNIST.inputs_dimTorchvisionClassificationDataModuleMNIST.outputs_dimTorchvisionClassificationDataModuleMNIST.targets_dimTorchvisionClassificationDataModuleMNIST.meanTorchvisionClassificationDataModuleMNIST.stdTorchvisionClassificationDataModuleMNIST.prepare_data()TorchvisionClassificationDataModuleMNIST.available_transformsTorchvisionClassificationDataModuleMNIST.available_test_augmentationsTorchvisionClassificationDataModuleMNIST.test_augmentations
TorchvisionClassificationDataModuleFashionMNISTTorchvisionClassificationDataModuleFashionMNIST.inputs_dimTorchvisionClassificationDataModuleFashionMNIST.outputs_dimTorchvisionClassificationDataModuleFashionMNIST.targets_dimTorchvisionClassificationDataModuleFashionMNIST.meanTorchvisionClassificationDataModuleFashionMNIST.stdTorchvisionClassificationDataModuleFashionMNIST.prepare_data()TorchvisionClassificationDataModuleFashionMNIST.available_transformsTorchvisionClassificationDataModuleFashionMNIST.available_test_augmentationsTorchvisionClassificationDataModuleFashionMNIST.test_augmentations
TorchvisionClassificationDataModuleSVHNTorchvisionClassificationDataModuleSVHN.inputs_dimTorchvisionClassificationDataModuleSVHN.outputs_dimTorchvisionClassificationDataModuleSVHN.targets_dimTorchvisionClassificationDataModuleSVHN.meanTorchvisionClassificationDataModuleSVHN.stdTorchvisionClassificationDataModuleSVHN.available_transformsTorchvisionClassificationDataModuleSVHN.available_test_augmentationsTorchvisionClassificationDataModuleSVHN.test_augmentations
TorchvisionClassificationDataModuleCIFAR10TorchvisionClassificationDataModuleCIFAR10.inputs_dimTorchvisionClassificationDataModuleCIFAR10.outputs_dimTorchvisionClassificationDataModuleCIFAR10.targets_dimTorchvisionClassificationDataModuleCIFAR10.meanTorchvisionClassificationDataModuleCIFAR10.stdTorchvisionClassificationDataModuleCIFAR10.available_transformsTorchvisionClassificationDataModuleCIFAR10.available_test_augmentationsTorchvisionClassificationDataModuleCIFAR10.test_augmentations
TorchvisionClassificationDataModuleCIFAR3TorchvisionClassificationDataModuleCIFAR3.inputs_dimTorchvisionClassificationDataModuleCIFAR3.outputs_dimTorchvisionClassificationDataModuleCIFAR3.targets_dimTorchvisionClassificationDataModuleCIFAR3.meanTorchvisionClassificationDataModuleCIFAR3.stdTorchvisionClassificationDataModuleCIFAR3.add_specific_args()TorchvisionClassificationDataModuleCIFAR3.available_transformsTorchvisionClassificationDataModuleCIFAR3.available_test_augmentationsTorchvisionClassificationDataModuleCIFAR3.test_augmentations
TorchvisionClassificationDataModuleCIFAR5TorchvisionClassificationDataModuleCIFAR5.inputs_dimTorchvisionClassificationDataModuleCIFAR5.outputs_dimTorchvisionClassificationDataModuleCIFAR5.targets_dimTorchvisionClassificationDataModuleCIFAR5.meanTorchvisionClassificationDataModuleCIFAR5.stdTorchvisionClassificationDataModuleCIFAR5.add_specific_args()TorchvisionClassificationDataModuleCIFAR5.available_transformsTorchvisionClassificationDataModuleCIFAR5.available_test_augmentationsTorchvisionClassificationDataModuleCIFAR5.test_augmentations
TorchvisionClassificationDataModuleCIFAR100TorchvisionClassificationDataModuleCIFAR100.inputs_dimTorchvisionClassificationDataModuleCIFAR100.outputs_dimTorchvisionClassificationDataModuleCIFAR100.targets_dimTorchvisionClassificationDataModuleCIFAR100.meanTorchvisionClassificationDataModuleCIFAR100.stdTorchvisionClassificationDataModuleCIFAR100.available_transformsTorchvisionClassificationDataModuleCIFAR100.available_test_augmentationsTorchvisionClassificationDataModuleCIFAR100.test_augmentations
MedMNISTClassificationDataModulePneumoniaMNISTClassificationDataModulePneumoniaMNISTClassificationDataModule.inputs_dimPneumoniaMNISTClassificationDataModule.outputs_dimPneumoniaMNISTClassificationDataModule.targets_dimPneumoniaMNISTClassificationDataModule.meanPneumoniaMNISTClassificationDataModule.stdPneumoniaMNISTClassificationDataModule.available_transformsPneumoniaMNISTClassificationDataModule.available_test_augmentationsPneumoniaMNISTClassificationDataModule.test_augmentations
BreastMNISTClassificationDataModuleBreastMNISTClassificationDataModule.inputs_dimBreastMNISTClassificationDataModule.outputs_dimBreastMNISTClassificationDataModule.targets_dimBreastMNISTClassificationDataModule.meanBreastMNISTClassificationDataModule.stdBreastMNISTClassificationDataModule.available_transformsBreastMNISTClassificationDataModule.available_test_augmentationsBreastMNISTClassificationDataModule.test_augmentations
DermaMNISTClassificationDataModuleDermaMNISTClassificationDataModule.inputs_dimDermaMNISTClassificationDataModule.outputs_dimDermaMNISTClassificationDataModule.targets_dimDermaMNISTClassificationDataModule.meanDermaMNISTClassificationDataModule.stdDermaMNISTClassificationDataModule.available_transformsDermaMNISTClassificationDataModule.available_test_augmentationsDermaMNISTClassificationDataModule.test_augmentations
BloodMNISTClassificationDataModuleBloodMNISTClassificationDataModule.inputs_dimBloodMNISTClassificationDataModule.outputs_dimBloodMNISTClassificationDataModule.targets_dimBloodMNISTClassificationDataModule.meanBloodMNISTClassificationDataModule.stdBloodMNISTClassificationDataModule.available_transformsBloodMNISTClassificationDataModule.available_test_augmentationsBloodMNISTClassificationDataModule.test_augmentations
ToyTwoMoonsClassificationDataModuleToyTwoMoonsClassificationDataModule.inputs_dimToyTwoMoonsClassificationDataModule.outputs_dimToyTwoMoonsClassificationDataModule.targets_dimToyTwoMoonsClassificationDataModule.taskToyTwoMoonsClassificationDataModule.inputs_dtypeToyTwoMoonsClassificationDataModule.outputs_dtypeToyTwoMoonsClassificationDataModule.prepare_data()ToyTwoMoonsClassificationDataModule.plot()ToyTwoMoonsClassificationDataModule.add_specific_args()
ToyTwoCirclesClassificationDataModuleUCIClassificationDataModuleBreastCancerUCIClassificationDataModuleAdultIncomeUCIClassificationDataModuleCarEvaluationUCIClassificationDataModuleCreditUCIClassificationDataModuleDermatologyUCIClassificationDataModuleECG5000ClassificationDataModuleECG5000ClassificationDataModule.outputs_dimECG5000ClassificationDataModule.outputs_dtypeECG5000ClassificationDataModule.inputs_dtypeECG5000ClassificationDataModule.targets_dimECG5000ClassificationDataModule.taskECG5000ClassificationDataModule.prepare_data()ECG5000ClassificationDataModule.plot()
- yamle.data.custom module
- yamle.data.datamodule module
BaseDataModuleBaseDataModule.data_shapeBaseDataModule.inputs_dimBaseDataModule.inputs_dtypeBaseDataModule.outputs_dimBaseDataModule.outputs_dtypeBaseDataModule.targets_dimBaseDataModule.taskBaseDataModule.ignore_indicesBaseDataModule.get_transform()BaseDataModule.get_transform_composition()BaseDataModule.train_transform()BaseDataModule.validation_transform()BaseDataModule.test_transform()BaseDataModule.test_augmentation()BaseDataModule.calibration_transform()BaseDataModule.train_target_transform()BaseDataModule.validation_target_transform()BaseDataModule.test_target_transform()BaseDataModule.calibration_target_transform()BaseDataModule.train_joint_transform()BaseDataModule.validation_joint_transform()BaseDataModule.test_joint_transform()BaseDataModule.calibration_joint_transform()BaseDataModule.train_dataset()BaseDataModule.train_dataset_size()BaseDataModule.train_dataloader()BaseDataModule.train_number_of_batches()BaseDataModule.validation_dataset()BaseDataModule.validation_dataset_size()BaseDataModule.validation_dataloader()BaseDataModule.validation_number_of_batches()BaseDataModule.test_dataset()BaseDataModule.test_dataset_size()BaseDataModule.test_dataloader()BaseDataModule.test_number_of_batches()BaseDataModule.calibration_dataset()BaseDataModule.calibration_dataset_size()BaseDataModule.calibration_dataloader()BaseDataModule.calibration_number_of_batches()BaseDataModule.total_dataset_size()BaseDataModule.sample_data()BaseDataModule.prepare_data()BaseDataModule.setup()BaseDataModule.plot()BaseDataModule.add_specific_args()
VisionDataModuleVisionClassificationDataModuleVisionRegressionDataModuleRealWorldDataModuleRealWorldDataModule.inputs_dtypeRealWorldDataModule.outputs_dtypeRealWorldDataModule.targets_dimRealWorldDataModule.available_test_augmentationsRealWorldDataModule.test_augmentationsRealWorldDataModule.get_transform()RealWorldDataModule.prepare_data()RealWorldDataModule.setup()RealWorldDataModule.available_transforms
RealWorldRegressionDataModuleRealWorldRegressionDataModule.inputs_dtypeRealWorldRegressionDataModule.outputs_dtypeRealWorldRegressionDataModule.targets_dimRealWorldRegressionDataModule.prepare_data()RealWorldRegressionDataModule.available_transformsRealWorldRegressionDataModule.available_test_augmentationsRealWorldRegressionDataModule.test_augmentations
RealWorldClassificationDataModuleRealWorldClassificationDataModule.inputs_dtypeRealWorldClassificationDataModule.outputs_dtypeRealWorldClassificationDataModule.targets_dimRealWorldClassificationDataModule.available_transformsRealWorldClassificationDataModule.available_test_augmentationsRealWorldClassificationDataModule.test_augmentations
- yamle.data.dataset_wrappers module
- yamle.data.depth module
DepthEstimationDataModuleDepthEstimationDataModule.meanDepthEstimationDataModule.stdDepthEstimationDataModule.inputs_dimDepthEstimationDataModule.outputs_dimDepthEstimationDataModule.taskDepthEstimationDataModule.inputs_dtypeDepthEstimationDataModule.outputs_dtypeDepthEstimationDataModule.available_test_augmentationsDepthEstimationDataModule.get_transform()DepthEstimationDataModule.plot()DepthEstimationDataModule.prepare_data()DepthEstimationDataModule.available_transformsDepthEstimationDataModule.test_augmentations
NYUv2DataModule
- yamle.data.reconstruction module
ECG5000ReconstructionDataModuleECG5000ReconstructionDataModule.outputs_dimECG5000ReconstructionDataModule.outputs_dtypeECG5000ReconstructionDataModule.inputs_dtypeECG5000ReconstructionDataModule.targets_dimECG5000ReconstructionDataModule.taskECG5000ReconstructionDataModule.prepare_data()ECG5000ReconstructionDataModule.plot()ECG5000ReconstructionDataModule.add_specific_args()ECG5000ReconstructionDataModule.available_transformsECG5000ReconstructionDataModule.available_test_augmentationsECG5000ReconstructionDataModule.test_augmentations
- yamle.data.regression module
ToyRegressionDataModuleToyRegressionDataModule.outputs_dimToyRegressionDataModule.targets_dimToyRegressionDataModule.outputs_dtypeToyRegressionDataModule.inputs_dimToyRegressionDataModule.taskToyRegressionDataModule.prepare_data()ToyRegressionDataModule.plot()ToyRegressionDataModule.add_specific_args()ToyRegressionDataModule.available_transformsToyRegressionDataModule.available_test_augmentationsToyRegressionDataModule.test_augmentations
UCIRegressionDataModuleUCIRegressionDataModule.taskUCIRegressionDataModule.outputs_dimUCIRegressionDataModule.outputs_dtypeUCIRegressionDataModule.inputs_dtypeUCIRegressionDataModule.plot()UCIRegressionDataModule.available_transformsUCIRegressionDataModule.available_test_augmentationsUCIRegressionDataModule.test_augmentations
MedMNISTRegressionDataModuleRetinaMNISTDataModuleConcreteUCIRegressionDataModuleEnergyUCIRegressionDataModuleBostonUCIRegressionDataModuleWineQualityUCIRegressionDataModuleYachtUCIRegressionDataModuleAbaloneUCIRegressionDataModuleTelemonitoringUCIRegressionDataModuleTimeSeriesRegressionDataModuleTemperatureTimeSeriesDataModuleWikiFaceDatasetTorchvisionRegressionDataModuleWikiFaceRegressionDataModuleWikiFaceRegressionDataModule.inputs_dimWikiFaceRegressionDataModule.outputs_dimWikiFaceRegressionDataModule.targets_dimWikiFaceRegressionDataModule.outputs_dtypeWikiFaceRegressionDataModule.meanWikiFaceRegressionDataModule.stdWikiFaceRegressionDataModule.available_transformsWikiFaceRegressionDataModule.available_test_augmentationsWikiFaceRegressionDataModule.test_augmentations
TorchvisionRotationRegressionDataModuleTorchvisionRotationRegressionDataModule.outputs_dimTorchvisionRotationRegressionDataModule.targets_dimTorchvisionRotationRegressionDataModule.outputs_dtypeTorchvisionRotationRegressionDataModule.prepare_data()TorchvisionRotationRegressionDataModule.add_specific_args()TorchvisionRotationRegressionDataModule.available_transformsTorchvisionRotationRegressionDataModule.available_test_augmentationsTorchvisionRotationRegressionDataModule.test_augmentations
TinyImageNetRotationRegressionDataModuleTinyImageNetRotationRegressionDataModule.inputs_dimTinyImageNetRotationRegressionDataModule.meanTinyImageNetRotationRegressionDataModule.stdTinyImageNetRotationRegressionDataModule.available_transformsTinyImageNetRotationRegressionDataModule.available_test_augmentationsTinyImageNetRotationRegressionDataModule.test_augmentations
TorchvisionRotationRegressionDataModuleMNISTTorchvisionRotationRegressionDataModuleMNIST.inputs_dimTorchvisionRotationRegressionDataModuleMNIST.meanTorchvisionRotationRegressionDataModuleMNIST.stdTorchvisionRotationRegressionDataModuleMNIST.prepare_data()TorchvisionRotationRegressionDataModuleMNIST.available_transformsTorchvisionRotationRegressionDataModuleMNIST.available_test_augmentationsTorchvisionRotationRegressionDataModuleMNIST.test_augmentations
TorchvisionRotationRegressionDataModuleFashionMNISTTorchvisionRotationRegressionDataModuleFashionMNIST.inputs_dimTorchvisionRotationRegressionDataModuleFashionMNIST.meanTorchvisionRotationRegressionDataModuleFashionMNIST.stdTorchvisionRotationRegressionDataModuleFashionMNIST.prepare_data()TorchvisionRotationRegressionDataModuleFashionMNIST.available_transformsTorchvisionRotationRegressionDataModuleFashionMNIST.available_test_augmentationsTorchvisionRotationRegressionDataModuleFashionMNIST.test_augmentations
TorchvisionRotationRegressionDataModuleSVHNTorchvisionRotationRegressionDataModuleSVHN.inputs_dimTorchvisionRotationRegressionDataModuleSVHN.meanTorchvisionRotationRegressionDataModuleSVHN.stdTorchvisionRotationRegressionDataModuleSVHN.available_transformsTorchvisionRotationRegressionDataModuleSVHN.available_test_augmentationsTorchvisionRotationRegressionDataModuleSVHN.test_augmentations
TorchvisionRotationRegressionDataModuleCIFAR10TorchvisionRotationRegressionDataModuleCIFAR10.inputs_dimTorchvisionRotationRegressionDataModuleCIFAR10.meanTorchvisionRotationRegressionDataModuleCIFAR10.stdTorchvisionRotationRegressionDataModuleCIFAR10.available_transformsTorchvisionRotationRegressionDataModuleCIFAR10.available_test_augmentationsTorchvisionRotationRegressionDataModuleCIFAR10.test_augmentations
TorchvisionRotationRegressionDataModuleCIFAR100TorchvisionRotationRegressionDataModuleCIFAR100.inputs_dimTorchvisionRotationRegressionDataModuleCIFAR100.meanTorchvisionRotationRegressionDataModuleCIFAR100.stdTorchvisionRotationRegressionDataModuleCIFAR100.available_transformsTorchvisionRotationRegressionDataModuleCIFAR100.available_test_augmentationsTorchvisionRotationRegressionDataModuleCIFAR100.test_augmentations
- yamle.data.segmentation module
TorchvisionSegmentationDataModuleTorchvisionSegmentationDataModule.meanTorchvisionSegmentationDataModule.stdTorchvisionSegmentationDataModule.inputs_dimTorchvisionSegmentationDataModule.outputs_dimTorchvisionSegmentationDataModule.taskTorchvisionSegmentationDataModule.inputs_dtypeTorchvisionSegmentationDataModule.outputs_dtypeTorchvisionSegmentationDataModule.available_test_augmentationsTorchvisionSegmentationDataModule.get_transform()TorchvisionSegmentationDataModule.plot()TorchvisionSegmentationDataModule.prepare_data()TorchvisionSegmentationDataModule.available_transformsTorchvisionSegmentationDataModule.test_augmentations
TorchvisionSegmentationDataModuleCityscapesTorchvisionSegmentationDataModuleCityscapes.inputs_dimTorchvisionSegmentationDataModuleCityscapes.meanTorchvisionSegmentationDataModuleCityscapes.stdTorchvisionSegmentationDataModuleCityscapes.ignore_indicesTorchvisionSegmentationDataModuleCityscapes.outputs_dimTorchvisionSegmentationDataModuleCityscapes.targets_dimTorchvisionSegmentationDataModuleCityscapes.plot()TorchvisionSegmentationDataModuleCityscapes.available_transformsTorchvisionSegmentationDataModuleCityscapes.available_test_augmentationsTorchvisionSegmentationDataModuleCityscapes.test_augmentations
- yamle.data.text module
TorchtextClassificationDataModuleTorchtextClassificationDataModule.meanTorchtextClassificationDataModule.stdTorchtextClassificationDataModule.taskTorchtextClassificationDataModule.inputs_dimTorchtextClassificationDataModule.inputs_dtypeTorchtextClassificationDataModule.outputs_dimTorchtextClassificationDataModule.outputs_dtypeTorchtextClassificationDataModule.prepare_data()TorchtextClassificationDataModule.plot()TorchtextClassificationDataModule.available_transformsTorchtextClassificationDataModule.available_test_augmentationsTorchtextClassificationDataModule.test_augmentations
TorchtextClassificationModelWikiText2TorchtextClassificationModelWikiText2.inputs_dimTorchtextClassificationModelWikiText2.outputs_dimTorchtextClassificationModelWikiText2.targets_dimTorchtextClassificationModelWikiText2.available_transformsTorchtextClassificationModelWikiText2.available_test_augmentationsTorchtextClassificationModelWikiText2.test_augmentations
TorchtextClassificationModelWikiText103TorchtextClassificationModelWikiText103.inputs_dimTorchtextClassificationModelWikiText103.outputs_dimTorchtextClassificationModelWikiText103.targets_dimTorchtextClassificationModelWikiText103.available_transformsTorchtextClassificationModelWikiText103.available_test_augmentationsTorchtextClassificationModelWikiText103.test_augmentations
TorchtextClassificationModelIMDBTorchtextClassificationModelIMDB.inputs_dimTorchtextClassificationModelIMDB.outputs_dimTorchtextClassificationModelIMDB.targets_dimTorchtextClassificationModelIMDB.available_transformsTorchtextClassificationModelIMDB.available_test_augmentationsTorchtextClassificationModelIMDB.test_augmentations
Shakespeare
- yamle.data.transforms module