ElasticNet Model

class qsar.models.elasticnet.ElasticnetModel(max_iter: int = 100000, random_state: int = 0, params=None)

Bases: BaselineModel

A class used to represent an ElasticNet model, inheriting from the Model class. This class specifically deals with the ElasticNet Regressor from the sklearn library.

optimize_hyperparameters(trial: Trial, df: DataFrame) → float

Optimizes the hyperparameters of the ElasticNet Regressor model using a trial from Optuna.

Parameters:

• trial (optuna.Trial) – The trial instance for hyperparameter optimization.

• df (pd.DataFrame) – The DataFrame used for training the model.

Returns:

The cross-validation score of the model.

Return type:

float