Add CLI option for adding known SRMs to results

Author: oshando

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/cli/CliRunner.java

@@ -65,6 +65,9 @@ public class CliRunner implements Callable<Integer> {
     @CommandLine.Option(names = {"-pt", "--prediction-threshold"}, description = {"Threshold for predicting categories"})
     private double predictionThreshold = 0.5;
 
+    @CommandLine.Option(names = {"-sr", "--known-srms"}, description = {"Add know SRMs from dataset"})
+    private boolean addKnownSrms = true;
+
     @CommandLine.Option(names = {"-ds", "--discovery"}, arity = "1..*", description = {"Select discovery for training set SRMs"})
     private List<String> discovery =  new ArrayList<>();//  Collections.singletonList("manual");
 
@@ -95,6 +98,7 @@ public SwanOptions initializeOptions(){
         options.setIterations(iterations);
         options.setTrainTestSplit(split);
         options.setPredictionThreshold(predictionThreshold);
+        options.setAddKnownSrms(addKnownSrms);
         options.setReduceAttributes(false);
         options.setDiscovery(discovery);
         options.setTimeLimit(timeLimit);

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/cli/SwanOptions.java

@@ -30,6 +30,7 @@ public class SwanOptions {
     private List<String> arffInstancesFiles;
     private List<String> discovery;
     private int timeLimit;
+    private boolean addKnownSrms;
 
     public SwanOptions(){
 
@@ -239,6 +240,14 @@ public void setTimeLimit(int timeLimit) {
         this.timeLimit = timeLimit;
     }
 
+    public boolean isAddKnownSrms() {
+        return addKnownSrms;
+    }
+
+    public void setAddKnownSrms(boolean addKnownSrms) {
+        this.addKnownSrms = addKnownSrms;
+    }
+
     @Override
     public String toString() {
         return "SwanOptions{" +