From 8ca062b80b7b3a9df0bf0927dfad1fe94f874b86 Mon Sep 17 00:00:00 2001 From: knikolaou Date: Thu, 16 May 2024 18:08:55 +0200 Subject: [PATCH] Include NTK measurement in imports --- papyrus/measurements/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py index f842afd..0c19e37 100644 --- a/papyrus/measurements/__init__.py +++ b/papyrus/measurements/__init__.py @@ -23,6 +23,7 @@ from papyrus.measurements.base_measurement import BaseMeasurement from papyrus.measurements.measurements import ( + NTK, Accuracy, Loss, NTKEigenvalues, @@ -41,4 +42,5 @@ NTKMagnitudeDistribution.__name__, Loss.__name__, Accuracy.__name__, + NTK.__name__, ]