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 subclass

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