diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py index ffdd395..92d80b7 100644 --- 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__, ]