From 94da42ed42db9f515e5bd552ae221a2d5775dc17 Mon Sep 17 00:00:00 2001 From: Michel Oleynik Date: Tue, 8 May 2018 13:48:27 +0200 Subject: [PATCH] Update NN runner for submission --- .../java/at/medunigraz/imi/bst/n2c2/BILSTMC3GFullRun.java | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/main/java/at/medunigraz/imi/bst/n2c2/BILSTMC3GFullRun.java b/src/main/java/at/medunigraz/imi/bst/n2c2/BILSTMC3GFullRun.java index 6b72edf..7d95144 100644 --- a/src/main/java/at/medunigraz/imi/bst/n2c2/BILSTMC3GFullRun.java +++ b/src/main/java/at/medunigraz/imi/bst/n2c2/BILSTMC3GFullRun.java @@ -26,18 +26,14 @@ public class BILSTMC3GFullRun { public static void main(String[] args) throws IOException { - // just needed for simulating test data for the moment - final File patienFolder = new File(Config.NN_SAMPLES_TRAINING); - // path to persisted BILSTMC3G model final String modelPath = Config.NN_MODELS; // path for saving prediction on test patients final String pathPredicted = Config.NN_SAMPLES_PREDICTED_BDT; - // simulate test data, put in here real test data - List patientsTraining = DatasetUtil.loadFromFolder(patienFolder); - List patientsTestBILSTMC3G = DatasetUtil.stripTags(patientsTraining); + // real test data + List patientsTestBILSTMC3G = DatasetUtil.loadFromFolder(new File("data/test")); // set port for monitoring neural networks Properties props = System.getProperties();