Function precision

tfjs-layers/src/metrics.ts:62–73

(yTrue: Tensor, yPred: Tensor)

}

export function precision(yTrue: Tensor, yPred: Tensor): Tensor {
  return tidy(() => {
    const tp = truePositives(yTrue, yPred);
    const fp = falsePositives(yTrue, yPred);

    const denominator = tfc.add(tp, fp);

    return tfc.cast(
        tfc.where(tfc.greater(denominator, 0), tfc.div(tp, denominator), 0),
        'float32');
  });
}

export function recall(yTrue: Tensor, yPred: Tensor): Tensor {
  return tidy(() => {