diff --git a/src/main/java/be/cylab/java/roc/Main.java b/src/main/java/be/cylab/java/roc/Main.java index 310eb4cb1692d3322aa1be3b54c8ca411183528e..3825f3bc0589259ed903a928b6096c18cf2d970a 100644 --- a/src/main/java/be/cylab/java/roc/Main.java +++ b/src/main/java/be/cylab/java/roc/Main.java @@ -49,6 +49,9 @@ public class Main { Roc roc = new Roc(solutions, true_alert); System.out.println(roc.computeAUC()); + List<RocCoordinates> rocCoordinates = roc.computeRocPoints(); + Utils.storeRocCoordinatesInCSVFile(rocCoordinates, "/home/alex/Desktop/Test.csv"); + } } diff --git a/src/main/java/be/cylab/java/roc/Roc.java b/src/main/java/be/cylab/java/roc/Roc.java index 233421f038daa1adbaf7428e57c86c9eed17b0e5..5967e580fd8b32c692557951bd2c1855c578e224 100644 --- a/src/main/java/be/cylab/java/roc/Roc.java +++ b/src/main/java/be/cylab/java/roc/Roc.java @@ -38,7 +38,7 @@ public class Roc { roc_points.add(new RocCoordinates(true_detection, false_alarm)); previous_score = point.getScore(); } - if (point.getScore() == 1) { + if (point.isTrueAlert()) { true_positive++; } else { false_positive++;