hyrax.verbs.reduction_algorithms.umap

Attributes

logger

Classes

UMAP	UMAP reduction implementation.

Module Contents

logger

class UMAP(config: dict, reduction_results=None)

Bases: hyrax.verbs.reduction_algorithms.algorithm_registry.ReductionAlgorithm

UMAP reduction implementation.

reducer

save_model(results_dir: pathlib.Path)

Save the fitted UMAP model to a pickle file.

Parameters:

results_dir (Path) – The directory where the model should be saved. The model will be saved as ‘umap.pickle’ in this directory.

load_model(expected_input_dim: int, model_path: pathlib.Path | str | None = None)

Load a pre-existing UMAP model from disk.

Parameters:

• expected_input_dim (int) – The expected number of input features for the loaded model.

• model_path (Path or str, optional) – The path to the file to load the model from. If not specified, method will look in the config for a default model path.

_validate_umap_model(reducer, expected_input_dim: int) → None

Validate the loaded UMAP model. Checks that the loaded object is a UMAP instance and that its input and output dimensions match the expected values.

Parameters:

• reducer (object) – The loaded model object to validate.

• expected_input_dim (int) – The expected number of input features for the loaded model.

Raises:

ValueError – If the loaded model is not a UMAP instance or if its input/output dimensions are incompatible.

fit(data_sample: numpy.ndarray)

Fit the UMAP model to a sample of inference data. The fitted model is stored in the instance variable self.reducer and can be used for transforming data.

Parameters:

data_sample (numpy.ndarray) – The data sample used to fit the model.

transform(args: dict, num_batches: int)

Transform data with a fitted UMAP model. Use parallel processing if specified in the config.

Parameters:

• args (dict) – A dictionary containing the data to be transformed.

• num_batches (int) – The total number of batches that the data is split into for transformation.