unbiased=True default for AngularCKA

wrongu · Apr 6, 2023 · 30fdeff · 30fdeff
1 parent 17c7957
commit 30fdeff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/repsim/metrics/angular_cka.py b/src/repsim/metrics/angular_cka.py
@@ -19,7 +19,7 @@ class AngularCKA(RepresentationMetricSpace, RiemannianSpace):
         Representations. NeurIPS. https://arxiv.org/abs/2110.14739
     """
 
-    def __init__(self, m, kernel=None, use_unbiased_hsic=False):
+    def __init__(self, m, kernel=None, use_unbiased_hsic=True):
         super().__init__(dim=m*(m+1)/2-1, shape=(m, m))
         self._kernel = kernel if kernel is not None else DEFAULT_KERNEL
         self._unbiased = use_unbiased_hsic