astra.models.regression

This module contains variables for instantiating regressors and their hyperparameter search grids.

astra.models.regression.REGRESSORS

A dictionary mapping model names to their corresponding scikit-learn regressor instances. [source]

Type:

dict[str, BaseEstimator]

astra.models.regression.REGRESSOR_PARAMS

A dictionary mapping model names to dictionaries of hyperparameters to search over. [source]

Type:

dict[str, dict[str, list]]

astra.models.regression.REGRESSOR_PARAMS_OPTUNA

A dictionary mapping model names to dictionaries of hyperparameters to search over using Optuna. [source]

Type:

dict[str, dict[str, optuna.distributions]]