Index

A

accelerator (gunz_ml.schemas.args.ArgsConfig attribute) AdaCos (class in gunz_ml.metrics.classification.adacos) AnalysisConfig (class in gunz_ml.schemas.analysis) AnalysisRunner (class in gunz_ml.analysis.runner) ANALYZE (gunz_ml.consts.Mode attribute)

apply_plot_filters() (in module gunz_ml.plots.optuna) ArcFace (class in gunz_ml.metrics.classification.arcface) args (gunz_ml.schemas.experiment.BaseExpConfig attribute) ArgsConfig (class in gunz_ml.schemas.args) ArtifactStore (class in gunz_ml.management.artifacts) AuditService (class in gunz_ml.management.monitor)

B

BaseExpConfig (class in gunz_ml.schemas.experiment) batch_size (gunz_ml.schemas.analysis.AnalysisConfig attribute)

build_db_uri() (in module gunz_ml.utils.connectivity) bulk_audit() (gunz_ml.management.monitor.AuditService method)

C

CATEGORICAL (gunz_ml.consts.OptunaType attribute) check_db_connection() (in module gunz_ml.utils.connectivity) check_mlflow_connection() (in module gunz_ml.utils) check_optuna_connection() (in module gunz_ml.utils) check_pruning_thresholds() (in module gunz_ml.integrations.optuna) CLEANUP (gunz_ml.consts.Mode attribute) cleanup_study() (gunz_ml.management.core.StudyCleaner method)

compute_confusion_matrix() (in module gunz_ml.analysis) (in module gunz_ml.analysis.errors) compute_gradient_norms() (in module gunz_ml.analysis) (in module gunz_ml.analysis.dynamics) CosFace (class in gunz_ml.metrics.classification.cosface) count_parameters() (in module gunz_ml.analysis) (in module gunz_ml.analysis.internals) create_study() (in module gunz_ml.integrations.optuna)

D

DatabaseConfig (class in gunz_ml.schemas.db) db (gunz_ml.schemas.experiment.BaseExpConfig attribute) db_name (gunz_ml.schemas.db.DatabaseConfig attribute) DeadNeuronMonitor (class in gunz_ml.callbacks.dead_neuron_monitor)

default (gunz_ml.schemas.experiment.BaseExpConfig attribute) download_file() (gunz_ml.management.artifacts.ArtifactStore method) driver (gunz_ml.schemas.db.DatabaseConfig attribute) dry_run (gunz_ml.schemas.args.ArgsConfig attribute)

E

EarlyStopping (class in gunz_ml.callbacks.early_stopping) EarlyStoppingOptuna (class in gunz_ml.callbacks.early_stopping_optuna) enable_dynamics (gunz_ml.schemas.analysis.AnalysisConfig attribute) enable_errors (gunz_ml.schemas.analysis.AnalysisConfig attribute) enable_internals (gunz_ml.schemas.analysis.AnalysisConfig attribute)

estimate_flops() (in module gunz_ml.analysis) (in module gunz_ml.analysis.internals) experiment_id (gunz_ml.management.tracking.RunInfo attribute) ExperimentDiscovery (class in gunz_ml.management.core) EXPORT (gunz_ml.consts.Mode attribute)

F

FAILED (gunz_ml.integrations.lightning.FinalizeStatus attribute) FastGC (class in gunz_ml.management.core) fetch_points() (gunz_ml.management.artifacts.ArtifactStore method) filter_studies() (gunz_ml.management.core.StudyDiscovery method) FinalizeStatus (class in gunz_ml.integrations.lightning) find_broken_trials() (gunz_ml.management.core.PruneService method) find_experiment() (gunz_ml.management.tracking.TrackingManager method)

find_pareto_frontier_fast() (in module gunz_ml.plots.optuna) FINISHED (gunz_ml.integrations.lightning.FinalizeStatus attribute) FLOAT (gunz_ml.consts.OptunaType attribute) forward() (gunz_ml.metrics.classification.adacos.AdaCos method) (gunz_ml.metrics.classification.arcface.ArcFace method) (gunz_ml.metrics.classification.cosface.CosFace method) (gunz_ml.metrics.classification.sphereface.SphereFace method) forward_hook() (gunz_ml.callbacks.prediction_dynamics.PredictionDynamicsMonitor method)

G

gen_partial_optimizer() (in module gunz_ml.optimizers) gen_partial_scheduler() (in module gunz_ml.optimizers) generate_plots() (in module gunz_ml.plots.optuna) get_all_studies() (gunz_ml.management.core.StudyDiscovery method) get_best_run() (gunz_ml.management.tracking.TrackingManager method) get_best_run_id() (gunz_ml.management.tracking.TrackingManager method) get_dict_param() (gunz_ml.schemas.run_cfg.RunConfig method) get_experiments() (gunz_ml.management.core.ExperimentDiscovery method) get_optuna_storage_url() (gunz_ml.management.core.ProjectMetadata method) get_param() (gunz_ml.schemas.run_cfg.RunConfig method) get_study_progress() (gunz_ml.management.monitor.AuditService method) get_study_stats() (gunz_ml.management.core.StudyMonitor method) get_top_k_errors() (in module gunz_ml.analysis) (in module gunz_ml.analysis.errors) GradientMonitor (class in gunz_ml.callbacks.gradient_monitor) gunz_ml module gunz_ml.analysis module gunz_ml.analysis.dynamics module gunz_ml.analysis.errors module gunz_ml.analysis.internals module gunz_ml.analysis.runner module gunz_ml.callbacks module gunz_ml.callbacks.dead_neuron_monitor module gunz_ml.callbacks.early_stopping module gunz_ml.callbacks.early_stopping_optuna module gunz_ml.callbacks.gradient_monitor module gunz_ml.callbacks.optuna module gunz_ml.callbacks.prediction_dynamics module gunz_ml.callbacks.weight_update_monitor module gunz_ml.configurator module gunz_ml.consts module gunz_ml.integrations module gunz_ml.integrations.hydra module gunz_ml.integrations.lightning module gunz_ml.integrations.mlflow module gunz_ml.integrations.optuna module gunz_ml.integrations.timm module gunz_ml.integrations.toml module gunz_ml.loggers module gunz_ml.loggers.modmlflow module gunz_ml.main module

gunz_ml.management module gunz_ml.management.artifacts module gunz_ml.management.core module gunz_ml.management.monitor module gunz_ml.management.scaffold module gunz_ml.management.tracking module gunz_ml.metrics module gunz_ml.metrics.classification module gunz_ml.metrics.classification.adacos module gunz_ml.metrics.classification.arcface module gunz_ml.metrics.classification.cosface module gunz_ml.metrics.classification.sphereface module gunz_ml.optimizers module gunz_ml.plots module gunz_ml.plots.categorical_distributions module gunz_ml.plots.contour module gunz_ml.plots.enhanced_scatter module gunz_ml.plots.optuna module gunz_ml.plots.pareto_by_category module gunz_ml.plots.pareto_continuous_color module gunz_ml.plots.utils module gunz_ml.schemas module gunz_ml.schemas.analysis module gunz_ml.schemas.args module gunz_ml.schemas.db module gunz_ml.schemas.experiment module gunz_ml.schemas.mlflow_cfg module gunz_ml.schemas.optuna_cfg module gunz_ml.schemas.optuna_cls module gunz_ml.schemas.run_cfg module gunz_ml.utils module gunz_ml.utils.connectivity module gunz_ml.utils.general module

H

high (gunz_ml.schemas.optuna_cls.OptunaFloatParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaIntParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaLogParamDC attribute)

host (gunz_ml.schemas.db.DatabaseConfig attribute) hparams (gunz_ml.schemas.optuna_cfg.OptunaConfig attribute)

I

ignore_pl_warnings() (in module gunz_ml.integrations.lightning) INIT (gunz_ml.consts.Mode attribute) init_callbacks() (in module gunz_ml.callbacks) init_hydra_and_check_config() (in module gunz_ml.integrations.hydra)

init_logger() (in module gunz_ml.loggers) init_trial_cfg() (in module gunz_ml.schemas) initialize() (gunz_ml.management.core.StudyInitializer method) INT (gunz_ml.consts.OptunaType attribute) is_job_active() (gunz_ml.management.core.SlurmTracker method)

L

layer_stats() (in module gunz_ml.analysis) (in module gunz_ml.analysis.internals) list_recursive() (gunz_ml.management.artifacts.ArtifactStore method) list_studies() (gunz_ml.management.tracking.TrackingManager method) LOAD (gunz_ml.consts.Mode attribute) load_mlf_exp() (in module gunz_ml.integrations.mlflow) load_numpy_from_mlf_run() (in module gunz_ml.integrations.mlflow) load_state_dict() (gunz_ml.callbacks.early_stopping.EarlyStopping method) load_study() (in module gunz_ml.integrations.optuna)

load_toml() (in module gunz_ml.integrations.toml) load_toml_as_dictconfig() (in module gunz_ml.integrations.toml) log_artifact() (gunz_ml.loggers.modmlflow.ModMLFlowLogger method) log_artifacts() (gunz_ml.loggers.modmlflow.ModMLFlowLogger method) log_hyperparams() (gunz_ml.loggers.modmlflow.ModMLFlowLogger method) log_hyperparams_metrics() (gunz_ml.loggers.modmlflow.ModMLFlowLogger method) LOGARITHM (gunz_ml.consts.OptunaType attribute) low (gunz_ml.schemas.optuna_cls.OptunaFloatParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaIntParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaLogParamDC attribute)

M

MAXIMIZE (gunz_ml.schemas.optuna_cfg.ObjectiveDirection attribute), [1] metrics (gunz_ml.management.tracking.RunInfo attribute) MINIMIZE (gunz_ml.schemas.optuna_cfg.ObjectiveDirection attribute), [1] mlf_run_to_dict() (in module gunz_ml.integrations.mlflow) MLFlowConfig (class in gunz_ml.schemas.mlflow_cfg) Mode (class in gunz_ml.consts) mode (gunz_ml.schemas.args.ArgsConfig attribute) mode_dict (gunz_ml.callbacks.early_stopping.EarlyStopping attribute) ModMLFlowLogger (class in gunz_ml.loggers.modmlflow) module gunz_ml gunz_ml.analysis gunz_ml.analysis.dynamics gunz_ml.analysis.errors gunz_ml.analysis.internals gunz_ml.analysis.runner gunz_ml.callbacks gunz_ml.callbacks.dead_neuron_monitor gunz_ml.callbacks.early_stopping gunz_ml.callbacks.early_stopping_optuna gunz_ml.callbacks.gradient_monitor gunz_ml.callbacks.optuna gunz_ml.callbacks.prediction_dynamics gunz_ml.callbacks.weight_update_monitor gunz_ml.configurator gunz_ml.consts gunz_ml.integrations gunz_ml.integrations.hydra gunz_ml.integrations.lightning gunz_ml.integrations.mlflow gunz_ml.integrations.optuna gunz_ml.integrations.timm gunz_ml.integrations.toml gunz_ml.loggers gunz_ml.loggers.modmlflow gunz_ml.main gunz_ml.management gunz_ml.management.artifacts gunz_ml.management.core gunz_ml.management.monitor gunz_ml.management.scaffold gunz_ml.management.tracking gunz_ml.metrics gunz_ml.metrics.classification gunz_ml.metrics.classification.adacos gunz_ml.metrics.classification.arcface gunz_ml.metrics.classification.cosface gunz_ml.metrics.classification.sphereface gunz_ml.optimizers gunz_ml.plots gunz_ml.plots.categorical_distributions gunz_ml.plots.contour gunz_ml.plots.enhanced_scatter gunz_ml.plots.optuna gunz_ml.plots.pareto_by_category gunz_ml.plots.pareto_continuous_color gunz_ml.plots.utils gunz_ml.schemas gunz_ml.schemas.analysis gunz_ml.schemas.args gunz_ml.schemas.db gunz_ml.schemas.experiment gunz_ml.schemas.mlflow_cfg gunz_ml.schemas.optuna_cfg gunz_ml.schemas.optuna_cls gunz_ml.schemas.run_cfg gunz_ml.utils gunz_ml.utils.connectivity gunz_ml.utils.general

monitor_op (gunz_ml.callbacks.early_stopping.EarlyStopping property)

N

name (gunz_ml.management.tracking.RunInfo attribute)

O

ObjectiveDirection (class in gunz_ml.schemas.optuna_cfg) objectives (gunz_ml.schemas.optuna_cls.OptunaConfigDC attribute) on_after_backward() (gunz_ml.callbacks.gradient_monitor.GradientMonitor method) on_before_optimizer_step() (gunz_ml.callbacks.weight_update_monitor.WeightUpdateMonitor method) on_train_end() (gunz_ml.callbacks.dead_neuron_monitor.DeadNeuronMonitor method) (gunz_ml.callbacks.prediction_dynamics.PredictionDynamicsMonitor method) on_train_epoch_end() (gunz_ml.callbacks.early_stopping.EarlyStopping method) (gunz_ml.callbacks.optuna.PyTorchLightningPruningCallback method) on_train_start() (gunz_ml.callbacks.dead_neuron_monitor.DeadNeuronMonitor method) (gunz_ml.callbacks.prediction_dynamics.PredictionDynamicsMonitor method) on_validation_end() (gunz_ml.callbacks.early_stopping.EarlyStopping method) (gunz_ml.callbacks.optuna.OptunaPruningCallback method) OPTIMIZE (gunz_ml.consts.Mode attribute) optimize (gunz_ml.schemas.optuna_cls.OptunaConfigDC attribute) optimize_cfg() (in module gunz_ml.integrations.optuna)

optimize_preproc_pipeline() (in module gunz_ml.schemas.optuna_cfg) optuna (gunz_ml.schemas.experiment.BaseExpConfig attribute) optuna_suggest() (in module gunz_ml.schemas.optuna_cfg) optuna_type (gunz_ml.schemas.experiment.BaseExpConfig attribute) OptunaCatParamDC (class in gunz_ml.schemas.optuna_cls) OptunaConfig (class in gunz_ml.schemas.optuna_cfg) OptunaConfigDC (class in gunz_ml.schemas.optuna_cls) OptunaFloatParamDC (class in gunz_ml.schemas.optuna_cls) OptunaIntParamDC (class in gunz_ml.schemas.optuna_cls) OptunaLogParamDC (class in gunz_ml.schemas.optuna_cls) OptunaPruningCallback (class in gunz_ml.callbacks.optuna) OptunaType (class in gunz_ml.consts) order_dict (gunz_ml.callbacks.early_stopping.EarlyStopping attribute) output_dir (gunz_ml.schemas.analysis.AnalysisConfig attribute) overwrite (gunz_ml.schemas.args.ArgsConfig attribute)

P

parameters (gunz_ml.schemas.optuna_cls.OptunaConfigDC attribute) params (gunz_ml.management.tracking.RunInfo attribute) params_cfg (gunz_ml.schemas.optuna_cfg.OptunaConfig attribute) password (gunz_ml.schemas.db.DatabaseConfig attribute) pl (gunz_ml.schemas.experiment.BaseExpConfig attribute) PLOT (gunz_ml.consts.Mode attribute) plot_categorical_distributions() (in module gunz_ml.plots.categorical_distributions) (in module gunz_ml.plots.optuna) plot_contour() (in module gunz_ml.plots.contour) (in module gunz_ml.plots.optuna) plot_enhanced_scatter() (in module gunz_ml.plots.enhanced_scatter) (in module gunz_ml.plots.optuna)

plot_pareto_continuous_color() (in module gunz_ml.plots.optuna) (in module gunz_ml.plots.pareto_continuous_color) plot_pareto_frontiers_by_category() (in module gunz_ml.plots.optuna) (in module gunz_ml.plots.pareto_by_category) port (gunz_ml.schemas.db.DatabaseConfig attribute) precision (gunz_ml.schemas.args.ArgsConfig attribute) PredictionDynamicsMonitor (class in gunz_ml.callbacks.prediction_dynamics) prepare_parameter_data() (in module gunz_ml.integrations.optuna) ProjectMetadata (class in gunz_ml.management.core) prune() (gunz_ml.management.core.PruneService method) PruneService (class in gunz_ml.management.core) purge_experiment() (gunz_ml.management.core.FastGC method) PyTorchLightningPruningCallback (class in gunz_ml.callbacks.optuna)

R

resolve_cfg() (in module gunz_ml.integrations.hydra) retrieve_mlf_exps() (in module gunz_ml.integrations.mlflow) retrieve_mlf_runs() (in module gunz_ml.integrations.mlflow) RUN (gunz_ml.consts.Mode attribute)

run() (gunz_ml.analysis.runner.AnalysisRunner method) run_id (gunz_ml.management.tracking.RunInfo attribute) RunConfig (class in gunz_ml.schemas.run_cfg) RunInfo (class in gunz_ml.management.tracking)

S

safe_set_experiment() (in module gunz_ml.integrations.mlflow) sanitize_metric_dataframe() (in module gunz_ml.integrations.optuna) sanitize_metric_names() (in module gunz_ml.integrations.optuna) sanitize_name() (in module gunz_ml.integrations.optuna) save_json (gunz_ml.schemas.analysis.AnalysisConfig attribute) save_plots (gunz_ml.schemas.analysis.AnalysisConfig attribute) scaffold_project() (in module gunz_ml.management.scaffold) search_by_metric() (gunz_ml.management.tracking.TrackingManager method) search_runs() (gunz_ml.management.tracking.TrackingManager method) select_objective_values() (in module gunz_ml.integrations.optuna) setup() (gunz_ml.callbacks.early_stopping.EarlyStopping method) SlurmTracker (class in gunz_ml.management.core) SphereFace (class in gunz_ml.metrics.classification.sphereface)

state_dict() (gunz_ml.callbacks.early_stopping.EarlyStopping method) state_key (gunz_ml.callbacks.early_stopping.EarlyStopping property) step (gunz_ml.schemas.optuna_cls.OptunaIntParamDC attribute) study_name (gunz_ml.schemas.args.ArgsConfig attribute) StudyCleaner (class in gunz_ml.management.core) StudyDiscovery (class in gunz_ml.management.core) StudyInitializer (class in gunz_ml.management.core) StudyMonitor (class in gunz_ml.management.core) StudyTruncator (class in gunz_ml.management.core) SUCCESS (gunz_ml.integrations.lightning.FinalizeStatus attribute) suggest_param() (gunz_ml.schemas.mlflow_cfg.MLFlowConfig method) (gunz_ml.schemas.optuna_cfg.OptunaConfig method) (gunz_ml.schemas.run_cfg.RunConfig method) suppress_pl_warnings() (in module gunz_ml.integrations.lightning)

T

tags (gunz_ml.management.tracking.RunInfo attribute) timeout (gunz_ml.schemas.db.DatabaseConfig attribute) TrackingManager (class in gunz_ml.management.tracking) trial (gunz_ml.schemas.optuna_cfg.OptunaConfig attribute) trials_to_dataframes() (in module gunz_ml.integrations.optuna)

truncate() (gunz_ml.management.core.StudyTruncator method) type (gunz_ml.schemas.optuna_cls.OptunaCatParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaFloatParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaIntParamDC attribute) (gunz_ml.schemas.optuna_cls.OptunaLogParamDC attribute)

U

uri_dict_to_str() (in module gunz_ml.utils.general)

user (gunz_ml.schemas.db.DatabaseConfig attribute)

V

vals (gunz_ml.schemas.optuna_cls.OptunaCatParamDC attribute)

W

WeightUpdateMonitor (class in gunz_ml.callbacks.weight_update_monitor)