metric

type HasExtraMetrics interface {
	GetExtraMetrics() []Value
}

type Metric interface {
	GetName() string
	Init()
	Reset()
	Compute(yTrue interface{}, yPred interface{}) Value
	ComputeFinal() Value
}

type SparseCategoricalAccuracy struct {
	Name       string
	Confidence float64
	Average    bool
	Precision  int
	// contains filtered or unexported fields
}

type SparseCategoricalFprAtTpr struct {
	Name                 string
	Tpr                  float64
	Attempts             int
	ComputeEveryNBatches int
	Precision            int
	ExtraMetrics         bool
	FinalConfidence      float64
	FinalTpr             float64
	// contains filtered or unexported fields
}

type SparseCategoricalTprAtFpr struct {
	Name                 string
	Fpr                  float64
	Attempts             int
	ComputeEveryNBatches int
	Precision            int
	FinalConfidence      float64
	FinalFpr             float64
	ExtraMetrics         bool
	// contains filtered or unexported fields
}

type Value struct {
	Name      string
	Value     float64
	Precision int
}