leaspy.models.utils.initialization.model_initialization.initialize_parameters

initialize_parameters(model, dataset, method='default') → tuple

Initialize the model’s group parameters given its name & the scores of all subjects.

Under-the-hood it calls an initialization function dedicated for the model:

• initialize_linear() (including when univariate)

• initialize_logistic() (including when univariate)

• initialize_logistic_parallel()

It is automatically called during Leaspy.fit().

Parameters:

modelAbstractModel

The model to initialize.

datasetDataset

Contains the individual scores.

methodstr

Must be one of:

• 'default': initialize at mean.

• 'random': initialize with a gaussian realization with same mean and variance.

Returns:

parametersdict [str, torch.Tensor]

Contains the initialized model’s group parameters.

Raises:

LeaspyInputError

If no initialization method is known for model type / method