From 720bd77c5fa3d7a2b2ac3bf527655267a9c3d92e Mon Sep 17 00:00:00 2001 From: "a.croix" <croix.alexandre@gmail.com> Date: Tue, 12 Mar 2019 18:05:15 +0100 Subject: [PATCH] Add method to store RocCoordinates in CSV file --- java-roc.iml | 6 ++++++ pom.xml | 9 +++++++- src/main/java/be/cylab/java/roc/Utils.java | 25 ++++++++++++++++++++++ 3 files changed, 39 insertions(+), 1 deletion(-) diff --git a/java-roc.iml b/java-roc.iml index ecf7c30..8d90f08 100644 --- a/java-roc.iml +++ b/java-roc.iml @@ -18,6 +18,12 @@ <orderEntry type="library" scope="TEST" name="Maven: org.junit.jupiter:junit-jupiter-params:5.3.1" level="project" /> <orderEntry type="library" scope="TEST" name="Maven: org.junit.jupiter:junit-jupiter-engine:5.3.1" level="project" /> <orderEntry type="library" scope="TEST" name="Maven: org.junit.platform:junit-platform-engine:1.3.1" level="project" /> + <orderEntry type="library" name="Maven: com.opencsv:opencsv:4.1" level="project" /> + <orderEntry type="library" name="Maven: org.apache.commons:commons-lang3:3.6" level="project" /> + <orderEntry type="library" name="Maven: org.apache.commons:commons-text:1.1" level="project" /> + <orderEntry type="library" name="Maven: commons-beanutils:commons-beanutils:1.9.3" level="project" /> + <orderEntry type="library" name="Maven: commons-logging:commons-logging:1.2" level="project" /> + <orderEntry type="library" name="Maven: commons-collections:commons-collections:3.2.2" level="project" /> <orderEntry type="library" name="Maven: com.owlike:genson:1.4" level="project" /> <orderEntry type="library" name="Maven: info.debatty:java-aggregation:0.4" level="project" /> <orderEntry type="module-library"> diff --git a/pom.xml b/pom.xml index 502c8c9..d0941e7 100644 --- a/pom.xml +++ b/pom.xml @@ -47,6 +47,13 @@ <version>5.3.1</version> <scope>test</scope> </dependency> + + <dependency> + <groupId>com.opencsv</groupId> + <artifactId>opencsv</artifactId> + <version>4.1</version> + </dependency> + <!-- Only for testing the last version of java-wowa-training --> <dependency> <groupId>com.owlike</groupId> @@ -68,7 +75,7 @@ <systemPath>${project.basedir}/../java-wowa-training/target/java-wowa-training-0.0.4-SNAPSHOT.jar</systemPath> </dependency> </dependencies> - <!-- ________________________________________________________________________________________________________________________> + <!-- ________________________________________________________________________________________________________________________--> <build> <plugins> <!-- leave this one first, to be sure we use this recent version --> diff --git a/src/main/java/be/cylab/java/roc/Utils.java b/src/main/java/be/cylab/java/roc/Utils.java index 06aa542..1ba4f54 100644 --- a/src/main/java/be/cylab/java/roc/Utils.java +++ b/src/main/java/be/cylab/java/roc/Utils.java @@ -1,5 +1,11 @@ package be.cylab.java.roc; +import com.opencsv.CSVWriter; + +import java.io.File; +import java.io.FileWriter; +import java.io.IOException; +import java.util.ArrayList; import java.util.List; public class Utils { @@ -23,4 +29,23 @@ public class Utils { } return negative_examples; } + + public static void storeRocCoordinatesInCSVFile(List<RocCoordinates> elements, String filepath) { + File file = new File(filepath); + try { + FileWriter outputfile = new FileWriter(file); + + CSVWriter writer = new CSVWriter(outputfile); + List<String[]> data = new ArrayList<>(); + for (RocCoordinates element : elements) { + String[] el = new String[] { Double.toString(element.getFalseAlarm()),Double.toString(element.getTrueDetection())}; + data.add(el); + } + writer.writeAll(data); + writer.close(); + } + catch (IOException e) { + e.printStackTrace(); + } + } } -- GitLab