cascade.core.DiscoverScheduler

class cascade.core.DiscoverScheduler(monitor, constraint, patience, tolerance=None, lam=None, alpha=None, gamma=None, **kwargs)

Bases: EarlyStopping

Hyperparameter scheduler for causal discovery

Parameters:

• monitor (str) – Loss to be monitored

• constraint (str) – Loss that specifies the constraint

• patience (int) – Number of checks with no improvement after which training will be stopped

• tolerance (float) – Maximal tolerance of constraint violation to end the scheduler

• lam (float) – Sparse penalty rate (\(\eta_\lambda\) in paper)

• alpha (float) – Acyclic penalty rate (\(\eta_\alpha\) in paper)

• gamma (float) – Kernel gradient rate (\(\eta_\gamma\))

• **kwargs – Additional keyword arguments are passed to EarlyStopping

Methods

on_validation_end

Main logic of the scheduler

Attributes

inf
state_key