From 29bf3b1af59f12e6b6877f7983490c0066f9d20f Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 22 Aug 2018 14:19:14 +0200
Subject: [PATCH] fixing invers_transform for WeightedRobustScaler
---
utils.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/utils.py b/utils.py
index ea98793..48f58f6 100644
--- a/utils.py
+++ b/utils.py
@@ -216,6 +216,15 @@ class WeightedRobustScaler(RobustScaler):
return super(WeightedRobustScaler, self).transform(X)
+ def inverse_transform(self, X):
+ if np.isnan(X).any():
+ X *= self.scale_
+ X += self.center_
+ return X
+ else:
+ return super(WeightedRobustScaler, self).inverse_transform(X)
+
+
def poisson_asimov_significance(s, ds, b, db):
"see `<http://www.pp.rhul.ac.uk/~cowan/stat/medsig/medsigNote.pdf>`_)"
db = np.sqrt(db**2+ds**2)
--
GitLab