Metric

abstract class Metric(reductionType: ReductionType)

Basic interface for all metric functions.

Constructors

Metric
Link copied to clipboard
fun Metric(reductionType: ReductionType)

Functions

apply
Link copied to clipboard
abstract fun apply(tf: Ops, yPred: Operand<Float>, yTrue: Operand<Float>, numberOfLabels: Operand<Float>?): Operand<Float>

Applies Metric to the yPred labels predicted by the model and known yTrue hidden during training.

Properties

reductionType
Link copied to clipboard
val reductionType: ReductionType

Reduction type. Should be defined in sub-class

Inheritors

Accuracy
Link copied to clipboard
MSE
Link copied to clipboard
MAE
Link copied to clipboard
MAPE
Link copied to clipboard
MSLE
Link copied to clipboard