Skip to content

Commit

Permalink
Update NN runner for submission
Browse files Browse the repository at this point in the history
  • Loading branch information
michelole committed May 8, 2018
1 parent ef9479b commit 94da42e
Showing 1 changed file with 2 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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<Patient> patientsTraining = DatasetUtil.loadFromFolder(patienFolder);
List<Patient> patientsTestBILSTMC3G = DatasetUtil.stripTags(patientsTraining);
// real test data
List<Patient> patientsTestBILSTMC3G = DatasetUtil.loadFromFolder(new File("data/test"));

// set port for monitoring neural networks
Properties props = System.getProperties();
Expand Down

0 comments on commit 94da42e

Please sign in to comment.