medianstopping

Documentation

Index

• type MedianPruner
  • func NewMedianPruner() *MedianPruner
• type PercentilePruner
  • func NewPercentilePruner(q float64) (*PercentilePruner, error)
  • func (p *PercentilePruner) Prune(study *goptuna.Study, trial goptuna.FrozenTrial) (bool, error)

Types

type MedianPruner

type MedianPruner struct {
	*PercentilePruner
}

MedianPruner implements a median stopping rule of Google Vizier. Prune if the trial's best intermediate result is worse than median of intermediate results of previous trials at the same step. See https://ai.google/research/pubs/pub46180 for more details.

func NewMedianPruner

func NewMedianPruner() *MedianPruner

NewMedianPruner is a constructor of MedianPruner.

type PercentilePruner

type PercentilePruner struct {
	Percentile     float64
	NStartUpTrials int
	NWarmUpSteps   int
}

PercentilePruner to keep the specified percentile of the trials.

func NewPercentilePruner

func NewPercentilePruner(q float64) (*PercentilePruner, error)

NewPercentilePruner is a constructor of percentile pruner

func (*PercentilePruner) Prune

func (p *PercentilePruner) Prune(study *goptuna.Study, trial goptuna.FrozenTrial) (bool, error)

Prune if the best intermediate value is in the bottom percentile among trials at the same step.