yamle.methods package#
- yamle.methods.method_factory(method_type)[source]#
This function is used to create a method instance based on the method type.
- Parameters:
method_type¶ (str) – The type of method to create.
- Return type:
Type[BaseMethod]
Submodules#
- yamle.methods.augmentation_classification module
AugmentationImageClassificationMethodCutOutImageClassificationMethodMixUpImageClassificationMethodCutMixImageClassificationMethodRandomErasingImageClassificationMethodRandomErasingImageClassificationMethod.add_specific_args()RandomErasingImageClassificationMethod.test_nameRandomErasingImageClassificationMethod.prepare_data_per_nodeRandomErasingImageClassificationMethod.allow_zero_length_dataloader_with_multiple_devicesRandomErasingImageClassificationMethod.training
- yamle.methods.be module
- yamle.methods.contrastive module
- yamle.methods.delta_uq module
- yamle.methods.dun module
- yamle.methods.early_exit module
- yamle.methods.ensemble module
- yamle.methods.evidential_regression module
- yamle.methods.gp module
- yamle.methods.laplace module
- yamle.methods.mcdropout module
- yamle.methods.method module
BaseMethodBaseMethod.tasksBaseMethod.on_fit_start()BaseMethod.on_test_start()BaseMethod.training_step()BaseMethod.validation_step()BaseMethod.test_step()BaseMethod.on_before_backward()BaseMethod.optimizer_step()BaseMethod.on_train_batch_end()BaseMethod.on_validation_batch_end()BaseMethod.on_test_batch_end()BaseMethod.on_train_epoch_start()BaseMethod.on_validation_epoch_start()BaseMethod.on_test_epoch_start()BaseMethod.on_train_epoch_end()BaseMethod.on_validation_epoch_end()BaseMethod.on_test_epoch_end()BaseMethod.on_after_model_load()BaseMethod.on_before_model_load()BaseMethod.on_before_method_load()BaseMethod.on_after_method_load()BaseMethod.reset_metrics()BaseMethod.get_parameters()BaseMethod.get_named_parameters()BaseMethod.configure_optimizers()BaseMethod.state_dict()BaseMethod.load_state_dict()BaseMethod.analyse()BaseMethod.backward()BaseMethod.on_after_backward()BaseMethod.evaluationBaseMethod.add_specific_args()
- yamle.methods.mimo module
MIMOMethodMIMMOMethodMixMoMethodMixVitMethodUnMixMoMethodUnMixMoMethod.test_nameUnMixMoMethod.prepare_data_per_nodeUnMixMoMethod.allow_zero_length_dataloader_with_multiple_devicesUnMixMoMethod.trainingUnMixMoMethod.on_train_epoch_start()UnMixMoMethod.on_train_epoch_end()UnMixMoMethod.state_dict()UnMixMoMethod.load_state_dict()UnMixMoMethod.add_specific_args()
DataMUXMethod
- yamle.methods.moe module
- yamle.methods.pe module
- yamle.methods.rbnn module
- yamle.methods.sgld module
- yamle.methods.sngp module
- yamle.methods.svi module
- yamle.methods.swag module
SWAGMethodSWAGMethod.state_dict()SWAGMethod.load_state_dict()SWAGMethod.on_train_epoch_end()SWAGMethod.on_validation_epoch_start()SWAGMethod.on_train_epoch_start()SWAGMethod.add_specific_args()SWAGMethod.test_nameSWAGMethod.prepare_data_per_nodeSWAGMethod.allow_zero_length_dataloader_with_multiple_devicesSWAGMethod.training
- yamle.methods.temperature_scaling module
- yamle.methods.uncertain_method module