TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfa.metrics

View source on GitHub

A module containing metrics that conform to Keras API.


cohens_kappa module: Implements Cohen's Kappa.

f_scores module: Implements F1 scores.

multilabel_confusion_matrix module: Implements Multi-label confusion matrix scores.

r_square module: Implements R^2 scores.


class CohenKappa: Computes Kappa score between two raters.

class F1Score: Computes F1 micro, macro or weighted based on the user's choice.

class FBetaScore: Computes F-Beta Score.

class MultiLabelConfusionMatrix: Computes Multi-label confusion matrix.

class RSquare: Compute R^2 score.