inverse transform

BorgwardtLab · Jan 12, 2024 · ef2f49f · ef2f49f
1 parent 9629ebe
commit ef2f49f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/proteinshake/transform.py b/proteinshake/transform.py
@@ -103,3 +103,9 @@ def stochastic_transform(self, Xy):
         for transform in self.stochastic_transforms:
             Xy = transform(Xy)
         return Xy
+
+    def inverse_transform(self, y):
+        for transform in self.transforms:
+            if hasattr(transform, "inverse_transform"):
+                y = transform.inverse_transform(y)
+        return y
diff --git a/tests/task.py b/tests/task.py
@@ -24,6 +24,9 @@ def inverse_transform(self, y):
                 print("X", X.shape)
                 print("y", y.shape)
                 break
+        print("EVALUATION")
+        metrics = task.evaluate(np.ones((10,)) * 3, np.ones((10,)) * 5)
+        print(metrics)
 
 
 if __name__ == "__main__":