immuneML.preprocessing.filters package

Submodules

immuneML.preprocessing.filters.ChainRepertoireFilter module

class immuneML.preprocessing.filters.ChainRepertoireFilter.ChainRepertoireFilter(keep_chain, result_path: Optional[pathlib.Path] = None)[source]

Bases: immuneML.preprocessing.filters.Filter.Filter

Removes all repertoires from the RepertoireDataset object which contain at least one sequence with chain different than “keep_chain” parameter. Note that this filter filters out repertoires, not individual sequences, and can thus only be applied to RepertoireDatasets.

Since the filter removes repertoires from the dataset (examples in machine learning setting), it cannot be used with TrainMLModel instruction. If you want to filter out repertoires including a given chain, see DatasetExport instruction with preprocessing.

Parameters

keep_chain (SequenceType) – Which chain should be kept.

YAML specification:

preprocessing_sequences:
    my_preprocessing:
        - my_filter:
            ChainRepertoireFilter:
                keep_chain: TRB
keeps_example_count() → bool[source]

Defines if the preprocessing can be run with TrainMLModel instruction; to be able to run with it, the preprocessing cannot change the number of examples in the dataset

process_dataset(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, result_path: pathlib.Path)[source]

immuneML.preprocessing.filters.ClonesPerRepertoireFilter module

class immuneML.preprocessing.filters.ClonesPerRepertoireFilter.ClonesPerRepertoireFilter(result_path: Optional[pathlib.Path] = None, lower_limit: int = - 1, upper_limit: int = - 1)[source]

Bases: immuneML.preprocessing.filters.Filter.Filter

Removes all repertoires from the RepertoireDataset, which contain fewer clonotypes than specified by the lower_limit, or more clonotypes than specified by the upper_limit. Note that this filter filters out repertoires, not individual sequences, and can thus only be applied to RepertoireDatasets.

Since the filter removes repertoires from the dataset (examples in machine learning setting), it cannot be used with TrainMLModel instruction. If you want to use this filter, see DatasetExport instruction with preprocessing.

Parameters

  • lower_limit (int) – The minimal inclusive lower limit for the number of clonotypes allowed in a repertoire.

  • upper_limit (int) – The maximal inclusive upper limit for the number of clonotypes allowed in a repertoire.

When no lower or upper limit is specified, or the value -1 is specified, the limit is ignored.

YAML specification:

preprocessing_sequences:
    my_preprocessing:
        - my_filter:
            ClonesPerRepertoireFilter:
                lower_limit: 100
                upper_limit: 100000
keeps_example_count() → bool[source]

Defines if the preprocessing can be run with TrainMLModel instruction; to be able to run with it, the preprocessing cannot change the number of examples in the dataset

process_dataset(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, result_path: Optional[pathlib.Path] = None)[source]

immuneML.preprocessing.filters.CountAggregationFunction module

class immuneML.preprocessing.filters.CountAggregationFunction.CountAggregationFunction(value)[source]

Bases: enum.Enum

An enumeration.

FIRST = 'first'
LAST = 'last'
MAX = 'max'
MEAN = 'mean'
MIN = 'min'
SUM = 'sum'

immuneML.preprocessing.filters.CountPerSequenceFilter module

class immuneML.preprocessing.filters.CountPerSequenceFilter.CountPerSequenceFilter(low_count_limit: int, remove_without_count: bool, remove_empty_repertoires: bool, batch_size: int, result_path: Optional[pathlib.Path] = None)[source]

Bases: immuneML.preprocessing.filters.Filter.Filter

Removes all sequences from a Repertoire when they have a count below low_count_limit, or sequences with no count value if remove_without_counts is True. This filter can be applied to Repertoires and RepertoireDatasets.

Parameters

  • low_count_limit (int) – The inclusive minimal count value in order to retain a given sequence.

  • remove_without_count (bool) – Whether the sequences without a reported count value should be removed.

  • remove_empty_repertoires (bool) – Whether repertoires without sequences should be removed.

  • true (Only has an effect when remove_without_count is also set to True. If this is) –

:param this preprocessing cannot be used with TrainMLModel: :param instruction: :param but only with DatasetExport instruction instead.: :param batch_size: number of repertoires that can be loaded at the same time (only affects the speed when applying this filter on a RepertoireDataset) :type batch_size: int

YAML specification:

preprocessing_sequences:
    my_preprocessing:
        - my_filter:
            CountPerSequenceFilter:
                remove_without_count: True
                remove_empty_repertoires: True
                low_count_limit: 3
                batch_size: 4
keeps_example_count() → bool[source]

Defines if the preprocessing can be run with TrainMLModel instruction; to be able to run with it, the preprocessing cannot change the number of examples in the dataset

process_dataset(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, result_path: pathlib.Path) → immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset[source]

immuneML.preprocessing.filters.DuplicateSequenceFilter module

class immuneML.preprocessing.filters.DuplicateSequenceFilter.DuplicateSequenceFilter(filter_sequence_type: immuneML.environment.SequenceType.SequenceType, batch_size: int, count_agg: immuneML.preprocessing.filters.CountAggregationFunction.CountAggregationFunction, result_path: Optional[pathlib.Path] = None)[source]

Bases: immuneML.preprocessing.filters.Filter.Filter

Collapses duplicate nucleotide or amino acid sequences within each repertoire in the given RepertoireDataset. This filter can be applied to Repertoires and RepertoireDatasets.

Sequences are considered duplicates if the following fields are identical:

  • amino acid or nucleotide sequence (whichever is specified)

  • v and j genes (note that the full field including subgroup + gene is used for matching, i.e. V1 and V1-1 are not considered duplicates)

  • chain

  • region type

For all other fields (the non-specified sequence type, custom lists, sequence identifier) only the first occurring value is kept.

Note that this means the count value of a sequence with a given sequence identifier might not be the same as before removing duplicates, unless count_agg = FIRST is used.

Parameters

  • filter_sequence_type (SequenceType) – Whether the sequences should be collapsed on the nucleotide or amino acid level. Valid options are defined by the SequenceType enum.

  • batch_size (int) – number of repertoires that can be loaded at the same time (only affects the speed)

  • count_agg (CountAggregationFunction) – determines how the sequence counts of duplicate sequences are aggregated. Valid options are defined by the CountAggregationFunction enum.

YAML specification:

preprocessing_sequences:
    my_preprocessing:
        - my_filter:
            DuplicateSequenceFilter:
                # required parameters:
                filter_sequence_type: AMINO_ACID
                # optional parameters (if not specified the values bellow will be used):
                batch_size: 4
                count_agg: SUM
classmethod build_object(**kwargs)[source]
static get_documentation()[source]
process_dataset(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, result_path: pathlib.Path) → immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset[source]

immuneML.preprocessing.filters.Filter module

class immuneML.preprocessing.filters.Filter.Filter(result_path: Optional[pathlib.Path] = None)[source]

Bases: immuneML.preprocessing.Preprocessor.Preprocessor, abc.ABC

check_dataset_not_empty(processed_dataset: immuneML.data_model.dataset.Dataset.Dataset, location='Filter')[source]

immuneML.preprocessing.filters.MetadataRepertoireFilter module

class immuneML.preprocessing.filters.MetadataRepertoireFilter.MetadataRepertoireFilter(criteria: dict, result_path: Optional[pathlib.Path] = None)[source]

Bases: immuneML.preprocessing.filters.Filter.Filter

Removes repertoires from a RepertoireDataset based on information stored in the metadata_file. Note that this filters out repertoires, not individual sequences, and can thus only be applied to RepertoireDatasets.

Since this filter changes the number of repertoires (examples for the machine learning task), it cannot be used with TrainMLModel instruction. To filter out repertoires, use preprocessing from the DatasetExport instruction that will create a new dataset ready to be used for training machine learning models.

Parameters

criteria (dict) – a nested dictionary that specifies the criteria for keeping certain columns. See CriteriaMatcher for a more detailed explanation.

YAML specification:

preprocessing_sequences:
    my_preprocessing:
        - my_filter:
            # Example filter that keeps repertoires with values greater than 1 in the "my_column_name" column of the metadata_file
            MetadataRepertoireFilter:
                type: GREATER_THAN
                value:
                    type: COLUMN
                    name: my_column_name
                threshold: 1
keeps_example_count() → bool[source]

Defines if the preprocessing can be run with TrainMLModel instruction; to be able to run with it, the preprocessing cannot change the number of examples in the dataset

process_dataset(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, result_path: pathlib.Path)[source]

Module contents