mutcleaner.cleaners.antitoxin_pard3_cleaner#
Functions
|
Clean Antitoxin dataset using configurable pipeline |
Create Antitoxin dataset cleaning pipeline |
Classes
|
Configuration class for Antitoxin dataset cleaner. |
- class mutcleaner.cleaners.antitoxin_pard3_cleaner.AntitoxinParD3CleanerConfig(pipeline_name='Antitoxin Pipeline', num_workers=16, validate_config=True, column_mapping=<factory>, filters=<factory>, type_conversions=<factory>, validate_mut_workers=16, process_workers=16, label_columns=<factory>, primary_label_column='label')[source]#
Bases:
BaseCleanerConfigConfiguration class for Antitoxin dataset cleaner. Inherits from BaseCleanerConfig and adds Antitoxin-specific configuration options.
Simply run mutcleaner.download_antitoxin_source_file() to download the dataset.
Alternatively, the raw Antitoxin file can be obtained from:
- Attributes:
- column_mappingDict[str, str]
Mapping from source to target column names
- filtersDict[str, Callable]
Filter conditions for data cleaning
- wt_sequencestr
Wildtype sequence for the dataset, used for mutation validation
- type_conversionsDict[str, str]
Data type conversion specifications
- validate_mut_workersint
Number of workers for mutation validation, set to -1 to use all available CPUs
- process_workersint
Number of workers for applying mutations to sequences, set to -1 to use all available CPUs
- label_columnsList[str]
List of score columns to process
- primary_label_columnstr
Primary score column for the dataset
- pipeline_namestr
Name of the cleaning pipeline
Methods
from_dict(config_dict)Create configuration object from dictionary
from_json(json_path)Load configuration from JSON file
get_summary()Get a human-readable summary of the configuration
merge(partial_config)Merge partial configuration with current configuration
to_dict([exclude_callables])Convert configuration to dictionary
to_json(json_path, **json_kwargs)Save configuration to JSON file
validate()Validate Antitoxin-specific configuration parameters
- column_mapping: Dict[str, str]#
- filters: Dict[str, Callable]#
- label_columns: List[str]#
- pipeline_name: str = 'Antitoxin Pipeline'#
- primary_label_column: str = 'label'#
- process_workers: int = 16#
- type_conversions: Dict[str, str]#
- validate()[source]#
Validate Antitoxin-specific configuration parameters
- Raises:
ValueError – If configuration is invalid
- Return type:
None
- validate_mut_workers: int = 16#
- wt_sequence = 'MANVEKMSVAVTPQQAAVMREAVEAGEYATASEIVREAVRDWLAKRELRHDDIRRLRQLWDEGKASGRPEPVDFDALRKEARQKLTEVPPNGR'#
- mutcleaner.cleaners.antitoxin_pard3_cleaner.clean_antitoxin_pard3_dataset(pipeline)[source]#
Clean Antitoxin dataset using configurable pipeline
- Parameters:
pipeline (
Pipeline) – Antitoxin dataset cleaning pipeline- Return type:
Tuple[Pipeline,MutationDataset]- Returns:
Pipeline: The cleaned pipeline - MutationDataset: The cleaned Antitoxin dataset
Examples
Use default configuration:
>>> pipeline = create_antitoxin_cleaner(df) # df is raw Antitoxin dataset file
Use partial configuration:
>>> pipeline = create_antitoxin_cleaner(df, config={ ... "validate_mut_workers": 8, ... })
Load configuration from file:
>>> pipeline = create_antitoxin_cleaner(df, config="config.json") >>> pipeline, dataset = clean_antitoxin_dataset(pipeline)
- mutcleaner.cleaners.antitoxin_pard3_cleaner.create_antitoxin_pard3_cleaner(dataset_or_path=None, config=None)[source]#
Create Antitoxin dataset cleaning pipeline
- Parameters:
dataset_or_path (
Union[DataFrame,str,Path,None]) – Raw dataset DataFrame or file path to Antitoxin dataset.config (
Union[AntitoxinParD3CleanerConfig,Dict[str,Any],str,Path,None]) – Configuration for the cleaning pipeline. Can be: - AntitoxinCleanerConfig object - Dictionary with configuration parameters (merged with defaults) - Path to JSON configuration file (str or Path) - None (uses default configuration)
- Return type:
- Returns:
Pipeline: The cleaning pipeline used
- Raises:
TypeError – If config has invalid type
ValueError – If configuration validation fails