Feature

feature_subsetting

This file contains implementation for model training on feature subsets.

class scalr.feature.feature_subsetting.FeatureSubsetting(feature_subsetsize: int, chunk_model_config: dict, chunk_model_train_config: dict, train_data: AnnData | AnnCollection, val_data: AnnData | AnnCollection, target: str, mappings: dict, dirpath: str | None = None, device: str = 'cpu')

Bases: object

Class for FeatureSubsetting.

It trains a model for each subsetted datasets, each containing feature_subsetsize genes as features.

get_updated_configs()

Returns updated configs.

train_chunked_models() → list[Module]

Trains a model for each subset data.

Returns:

List of models for each subset.

Return type:

list[nn.Module]