Skip to content

Commit 62af5ca

Browse files
committed
Rename and refactor handler for features
1 parent 7589543 commit 62af5ca

File tree

3 files changed

+384
-373
lines changed

3 files changed

+384
-373
lines changed

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/SwanPipeline.java

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,11 @@
22

33
import com.fasterxml.jackson.databind.ObjectMapper;
44
import de.fraunhofer.iem.swan.cli.SwanOptions;
5-
import de.fraunhofer.iem.swan.data.Category;
6-
import de.fraunhofer.iem.swan.data.Method;
7-
import de.fraunhofer.iem.swan.features.InstancesHandler;
8-
import de.fraunhofer.iem.swan.features.code.FeatureHandler;
9-
import de.fraunhofer.iem.swan.features.code.soot.Loader;
10-
import de.fraunhofer.iem.swan.features.doc.DocFeatureHandler;
11-
import de.fraunhofer.iem.swan.io.dataset.Parser;
12-
import de.fraunhofer.iem.swan.io.dataset.Writer;
13-
import de.fraunhofer.iem.swan.model.Learner;
5+
import de.fraunhofer.iem.swan.features.FeaturesHandler;
6+
import de.fraunhofer.iem.swan.features.code.soot.SourceFileLoader;
7+
import de.fraunhofer.iem.swan.io.dataset.SrmList;
8+
import de.fraunhofer.iem.swan.io.dataset.SrmListUtils;
9+
import de.fraunhofer.iem.swan.model.ModelEvaluator;
1410
import de.fraunhofer.iem.swan.util.Util;
1511
import org.apache.commons.lang3.StringUtils;
1612
import org.slf4j.Logger;
@@ -64,8 +60,9 @@ public void run() throws IOException, InterruptedException {
6460
SourceFileLoader testDataset = new SourceFileLoader(options.getTestDataDir());
6561
testDataset.load(dataset.getMethods());
6662

67-
// Save data from last classification.
68-
loader.resetMethods();
63+
//Initialize and populate features
64+
FeaturesHandler featuresHandler = new FeaturesHandler(dataset, testDataset, options);
65+
featuresHandler.createFeatures();
6966

7067
// Cache the methods from the second test set.
7168
loader.pruneNone();

0 commit comments

Comments
 (0)