functional

consistency_dann_from_config(config, pre_trainer_kwargs=None, trainer_kwargs=None)

Construct a Consistency DANN approach from a configuration. The configuration can be created by calling get_consistency_dann_config.

Parameters:

Name	Type	Description	Default
config	DictConfig	The Consistency DANN config.	required
pre_trainer_kwargs	Optional[Dict[str, Any]]	Overrides for the pre-training trainer class.	None
trainer_kwargs	Optional[Dict[str, Any]]	Overrides for the main trainer class.	None

Returns: pre: The data module, approach and trainer for the pre-training stage main: The data module, approach, domain discriminator and trainer for the main stage

get_consistency_dann(dataset, source_fd, target_fd, pre_trainer_kwargs=None, trainer_kwargs=None)

Construct a Consistency DANN approach with the original hyperparameters.

Examples:

>>> import rul_adapt
>>> pre, main = rul_adapt.construct.get_consistency_dann("cmapss", 3, 1)
>>> pre_dm, pre_approach, pre_trainer = pre
>>> dm, approach, domain_disc, trainer = main

Parameters:

Name	Type	Description	Default
dataset	Literal['cmapss', 'xjtu-sy']	The name of the dataset, either cmapss or xjtu-sy.	required
source_fd	int	The source FD of CMAPSS.	required
target_fd	int	The target FD of CMAPSS.	required
pre_trainer_kwargs	Optional[Dict[str, Any]]	Overrides for the pre-training trainer class.	None
trainer_kwargs	Optional[Dict[str, Any]]	Overrides for the main trainer class.	None

Returns: pre: The data module, approach and trainer for the pre-training stage main: The data module, approach, domain discriminator and trainer for the main stage

get_consistency_dann_config(dataset, source_fd, target_fd)

Get a configuration for the Consistency DANN approach.

The configuration can be modified and fed to consistency_dann_from_config to create the approach.

Parameters:

Name	Type	Description	Default
dataset	Literal['cmapss', 'xjtu-sy']	The name of the dataset, either cmapss or xjtu-sy.	required
source_fd	int	The source FD of CMAPSS.	required
target_fd	int	The target FD of CMAPSS.	required

Returns: The Consistency DANN configuration.