|
2 | 2 |
|
3 | 3 | import com.fasterxml.jackson.databind.ObjectMapper; |
4 | 4 | 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; |
14 | 10 | import de.fraunhofer.iem.swan.util.Util; |
15 | 11 | import org.apache.commons.lang3.StringUtils; |
16 | 12 | import org.slf4j.Logger; |
@@ -64,8 +60,9 @@ public void run() throws IOException, InterruptedException { |
64 | 60 | SourceFileLoader testDataset = new SourceFileLoader(options.getTestDataDir()); |
65 | 61 | testDataset.load(dataset.getMethods()); |
66 | 62 |
|
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(); |
69 | 66 |
|
70 | 67 | // Cache the methods from the second test set. |
71 | 68 | loader.pruneNone(); |
|
0 commit comments