add new observables to init

zincware · Aug 15, 2024 · 66b2e68 · 66b2e68
1 parent 456ea29
commit 66b2e68
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py
@@ -25,9 +25,11 @@
 from papyrus.measurements.measurements import (
     NTK,
     Accuracy,
+    LabelNTKAlignment,
     Loss,
     LossDerivative,
     NTKEigenvalues,
+    NTKEigenvectorAlignment,
     NTKEntropy,
     NTKMagnitudeDistribution,
     NTKSelfEntropy,
@@ -45,4 +47,6 @@
     Accuracy.__name__,
     NTK.__name__,
     LossDerivative.__name__,
+    LabelNTKAlignment.__name__,
+    NTKEigenvectorAlignment.__name__,
 ]