Add new CLI options for model evaluation

Author: oshando

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

@@ -38,6 +38,14 @@ class CliRunner implements Callable<Integer> {
     @CommandLine.Option(names = {"-", "--arff-data"}, description = {"Export training ARFF files"})
     private boolean exportArffData = true;
 
+    @CommandLine.Option(names = {"-doc", "--documented"}, description = {"Use only methods with Javadoc"})
+    private boolean isDocumented = true;
+
+    @CommandLine.Option(names = {"-i", "--iterations"}, description = {"Number of iterations for training"})
+    private int iterations = 10;
+
+    @CommandLine.Option(names = {"-sp", "--training-split"}, description = {"Percentage for training"})
+    private double split = 0.7;
 
     @Override
     public Integer call() throws Exception {
@@ -50,7 +58,10 @@ public Integer call() throws Exception {
                 learningMode,
                 srmClasses,
                 cweClasses,
-                exportArffData);
+                exportArffData,
+                isDocumented,
+                iterations,
+                split);
 
         return new SwanCli().run(options);
     }

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

@@ -16,6 +16,9 @@ public class SwanOptions {
     private List<String> srmClasses;
     private List<String> cweClasses;
     private boolean exportArffData;
+    private boolean isDocumented;
+    private int iterations;
+    private double trainTestSplit;
 
     public SwanOptions(String testData, String trainData, String datasetJson, String outputDir, String featureSet,
                        String learningMode, List<String> srmClasses, List<String> cweClasses, boolean exportArffData) {
@@ -28,6 +31,9 @@ public SwanOptions(String testData, String trainData, String datasetJson, String
         this.srmClasses = srmClasses;
         this.cweClasses = cweClasses;
         this.exportArffData = exportArffData;
+        this.isDocumented = isDocumented;
+        this.iterations = iterations;
+        this.trainTestSplit = trainTestSplit;
     }
 
     public String getTestData() {
@@ -102,6 +108,30 @@ public void setExportArffData(boolean exportArffData) {
         this.exportArffData = exportArffData;
     }
 
+    public boolean isDocumented() {
+        return isDocumented;
+    }
+
+    public void setDocumented(boolean documented) {
+        this.isDocumented = documented;
+    }
+
+    public int getIterations() {
+        return iterations;
+    }
+
+    public void setIterations(int iterations) {
+        this.iterations = iterations;
+    }
+
+    public double getTrainTestSplit() {
+        return trainTestSplit;
+    }
+
+    public void setTrainTestSplit(double trainTestSplit) {
+        this.trainTestSplit = trainTestSplit;
+    }
+
     @Override
     public String toString() {
         return "SwanOptions{" +
@@ -114,6 +144,7 @@ public String toString() {
                 ", srmClasses=" + srmClasses +
                 ", cweClasses=" + cweClasses +
                 ", exportArffData=" + exportArffData +
+                ", isDocumented=" + isDocumented +
                 '}';
     }
 }