Add attribute selection, discovery and time limit CLI options

Author: oshando

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

@@ -50,6 +50,9 @@ public class CliRunner implements Callable<Integer> {
     @CommandLine.Option(names = {"-doc", "--documented"}, description = {"Use only methods with Javadoc"})
     private boolean isDocumented = true;
 
+    @CommandLine.Option(names = {"-att", "--attribute-selection"}, description = {"Use attribute selection"})
+    private boolean reduceAttributes = false;
+
     @CommandLine.Option(names = {"-i", "--iterations"}, description = {"Number of iterations for training"})
     private int iterations = 10;
 
@@ -62,6 +65,12 @@ 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 = {"-ds", "--discovery"}, arity = "1..*", description = {"Select discovery for training set SRMs"})
+    private List<String> discovery =  Collections.singletonList("manual");
+
+    @CommandLine.Option(names = {"-tl", "--timelimit"}, description = {"Time (minutes) to execute operation "})
+    private int timeLimit = 1;
+
     public SwanOptions initializeOptions(){
 
         SwanOptions options = new SwanOptions(testDataDir,
@@ -81,6 +90,9 @@ public SwanOptions initializeOptions(){
         options.setInstances(arffInstancesFiles);
         options.setTrainDataSourceDir(trainDataDirSource);
         options.setTestDataSourceDir(testDataSourceDir);
+        options.setReduceAttributes(false);
+        options.setDiscovery(discovery);
+        options.setTimeLimit(timeLimit);
 
         return options;
     }

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

@@ -22,11 +22,14 @@ public class SwanOptions {
     private List<String> cweClasses;
     private boolean exportArffData;
     private boolean isDocumented;
+    private boolean reduceAttributes;
     private int iterations;
     private double trainTestSplit;
     private String phase;
     private double predictionThreshold;
     private List<String> arffInstancesFiles;
+    private List<String> discovery;
+    private int timeLimit;
 
     public SwanOptions(String testDataDir, String trainDataDir, String datasetJson, String outputDir,
                        List<String> featureSet, String toolkit, List<String> srmClasses,
@@ -136,6 +139,14 @@ public boolean isExportArffData() {
         return exportArffData;
     }
 
+    public boolean isReduceAttributes() {
+        return reduceAttributes;
+    }
+
+    public void setReduceAttributes(boolean reduceAttributes) {
+        this.reduceAttributes = reduceAttributes;
+    }
+
     public void setExportArffData(boolean exportArffData) {
         this.exportArffData = exportArffData;
     }
@@ -209,6 +220,22 @@ public void setTrainDataSourceDir(String trainDataSourceDir) {
         this.trainDataSourceDir = trainDataSourceDir;
     }
 
+    public List<String> getDiscovery() {
+        return discovery;
+    }
+
+    public void setDiscovery(List<String> discovery) {
+        this.discovery = discovery;
+    }
+
+    public int getTimeLimit() {
+        return timeLimit;
+    }
+
+    public void setTimeLimit(int timeLimit) {
+        this.timeLimit = timeLimit;
+    }
+
     @Override
     public String toString() {
         return "SwanOptions{" +