1 files changed, 4 insertions, 6 deletions
diff --git a/npdistortion.h b/npdistortion.h
index a84f63a..a70616d 100644
--- a/npdistortion.h
+++ b/npdistortion.h
@@ -1,15 +1,13 @@
 #ifndef NPDISTORTION_H
 #define NPDISTORTION_H
 
-#include "distortionobserver.h"
+#include "distortionmeasure.h"
 
-class NPDistortion : public DistortionObserver
+class NPDistortion : public DistortionMeasure
 {
 public:
-    NPDistortion(const arma::mat &X, const arma::uvec &sampleIndices, int k = 10);
-
-protected:
-    arma::vec measureFunc(const arma::mat &distA, const arma::mat &distB);
+    NPDistortion(int k = 10);
+    arma::vec measure(const arma::mat &distA, const arma::mat &distB);
 
 private:
     int m_k;