Module epiclass.core.lgbm

Module to define how LightGBM is handled.

Functions

def print_last_trial(study, trial)

Optuna callback to always print more information on last finished trial.

def tune_lgbm(ea_handler: EpiAtlasFoldFactory, logdir: pathlib.Path)

It takes an EpiAtlasFoldFactory object and a log directory, and it tunes the hyperparameters of a LightGBM classifier using Optuna.

Args

ea_handler : EpiAtlasFoldFactory

Dataset splits creator.

logdir : Path

The directory where the results will be saved.

Returns

The Optuna study object.