earlystopping

Documentation

Index

• Constants
• type EarlyStoppingService
• type MedianStoppingParam
• type MedianStoppingRule
  • func NewMedianStoppingRule() *MedianStoppingRule
  • func (m *MedianStoppingRule) GetShouldStopWorkers(ctx context.Context, in *api.GetShouldStopWorkersRequest) (*api.GetShouldStopWorkersReply, error)

Constants

const (
	// DefaultPort is the port to serve the earlystopping service.
	DefaultPort = "0.0.0.0:6789"
)

Types

type EarlyStoppingService

type EarlyStoppingService interface {
	GetEarlyStoppingParameters(ctx context.Context, in *api.GetEarlyStoppingParametersRequest) (api.GetEarlyStoppingParametersReply, error)
}

EarlyStoppingService is the interface for earlystopping service.

type MedianStoppingParam

type MedianStoppingParam struct {
	LeastStep  int
	Margin     float64
	EvalMetric string
	BurnIn     int
}

type MedianStoppingRule

type MedianStoppingRule struct {
	// contains filtered or unexported fields
}

func NewMedianStoppingRule

func NewMedianStoppingRule() *MedianStoppingRule

func (*MedianStoppingRule) GetShouldStopWorkers

func (m *MedianStoppingRule) GetShouldStopWorkers(ctx context.Context, in *api.GetShouldStopWorkersRequest) (*api.GetShouldStopWorkersReply, error)