astra.models.classification

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

astra.models.classification.CLASSIFIERS

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

Type:

dict[str, BaseEstimator]

astra.models.classification.CLASSIFIER_PARAMS

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

Type:

dict[str, dict[str, list]]

astra.models.classification.CLASSIFIER_PARAMS_OPTUNA

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

Type:

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

astra.models.classification.NON_PROBABILISTIC_MODELS

A list of model names that do not have a predict_proba method. [source]

Type:

list[str]