ForeTiS.model.elasticnet

Module Contents

Classes

ElasticNet

Implementation of a class for ElasticNet.
class ForeTiS.model.elasticnet.ElasticNet(optuna_trial, datasets, featureset_name, pca_transform, target_column, optimize_featureset)

Bases: ForeTiS.model._sklearn_model.SklearnModel

Implementation of a class for ElasticNet.

See BaseModel for more information on the attributes.

Parameters:

  • optuna_trial (optuna.trial.Trial)

  • datasets (list)

  • featureset_name (str)

  • pca_transform (bool)

  • target_column (str)

  • optimize_featureset (bool)

define_model()

Definition of the actual prediction model.

See BaseModel for more information.

Return type:

sklearn.linear_model.ElasticNet

define_hyperparams_to_tune()

See BaseModel for more information on the format.

Return type:

dict