CatBoost Model
- class qsar.models.catboost.CatboostModel(max_iter: int = 100000, random_state: int = 0, params=None)
Bases:
BaselineModel
A class used to represent a CatboostModel which inherits from the Model class. This class is specific for handling CatBoost Regressor models.
- optimize_hyperparameters(trial: Trial, df: DataFrame) float
Optimizes the hyperparameters of the CatBoost Regressor model using a trial from Optuna.
- Parameters:
trial (Trial) – the trial for hyperparameter optimization
df (pd.DataFrame) – the dataframe used for training the model
- Returns:
the cross validation score of the model
- Return type:
float