Add Kendall's Tau-c, Spearman's Rank, Pearson's Correlation metrics (#…

…2741) * renaming kendalls_tau to streaming_correlations * Fix keras import * Fix test function naming * Fixing commit history to improve review experience
tensorflow · Aug 5, 2022 · 5759e3d · 5759e3d
1 parent c7c40a0
commit 5759e3d
Show file tree

Hide file tree

Showing 5 changed files with 478 additions and 304 deletions.
diff --git a/tensorflow_addons/metrics/__init__.py b/tensorflow_addons/metrics/__init__.py
@@ -31,4 +31,9 @@
 from tensorflow_addons.metrics.r_square import RSquare
 from tensorflow_addons.metrics.geometric_mean import GeometricMean
 from tensorflow_addons.metrics.harmonic_mean import HarmonicMean
-from tensorflow_addons.metrics.kendalls_tau import KendallsTau
+from tensorflow_addons.metrics.streaming_correlations import (
+    KendallsTauB,
+    KendallsTauC,
+    PearsonsCorrelation,
+    SpearmansRank,
+)
diff --git a/tensorflow_addons/metrics/kendalls_tau.py b/tensorflow_addons/metrics/kendalls_tau.py