fix roc curves

heidelberg-hepml · Mar 16, 2023 · 865722d · 865722d
1 parent 48db8f6
commit 865722d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/plots.py b/src/plots.py
@@ -140,7 +140,7 @@ def plot_roc(self, file: str):
         Args:
             file: Output file name
         """
-        scores = np.concatenate((self.weights_true, self.weights_fake), axis=0)
+        scores = -np.concatenate((1/self.weights_true, self.weights_fake), axis=0)
         labels = np.concatenate((
             np.ones_like(self.weights_true),
             np.zeros_like(self.weights_fake)