Skip to content

Commit adb8f6e

Browse files
committed
Include origin of instances in file names
1 parent 86d360e commit adb8f6e

File tree

4 files changed

+8
-9
lines changed

4 files changed

+8
-9
lines changed

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/features/MekaFeatureSet.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ public Instances convertToMekaInstances(Instances instances) {
107107
output = Filter.useFilter(instances, filter);
108108
output.setRelationName("swan-srm:" + output.relationName());
109109

110-
Util.exportInstancesToArff(output);
110+
Util.exportInstancesToArff(output, "meka");
111111
} catch (Exception e) {
112112
e.printStackTrace();
113113
}

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/features/WekaFeatureSet.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ public void createFeatures() {
3636
String instanceName = category.getId().toLowerCase() + "-train-instances";
3737
Instances trainInstances = createInstances(trainAttributes, dataset.getTrainMethods(), Collections.singleton(category));
3838
this.instances.put(category.getId().toLowerCase(), trainInstances);
39-
Util.exportInstancesToArff(trainInstances);
39+
Util.exportInstancesToArff(trainInstances, "weka");
4040

4141
//Create and set attributes for the test instances.
4242
/*ArrayList<Attribute> testAttributes = createAttributes(getCategories(category), testData.getMethods(), featureSets);

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/model/toolkit/MLPlan.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ public MLPlan() {
4949
*/
5050
public HashMap<String, ArrayList<Double>> evaluateDataset(Instances instances1) {
5151

52-
String arffFilePath = Util.exportInstancesToArff(instances1);
52+
String arffFilePath = Util.exportInstancesToArff(instances1, "mlplan");
5353
ArffDatasetAdapter arffDatasetAdapter = new ArffDatasetAdapter();
5454

5555
String mClass = Util.getClassName(instances1);
@@ -148,7 +148,7 @@ public HashMap<String, ArrayList<Double>> evaluateDataset(Instances instances1)
148148
public void evaluateDataset(Instances instances, int k) {
149149

150150
//arffFilePath = "swan/swan_core/src/main/resources/waveform.arff";
151-
String arffFilePath = Util.exportInstancesToArff(instances);
151+
String arffFilePath = Util.exportInstancesToArff(instances, "mlplan");
152152
ArffDatasetAdapter arffDatasetAdapter = new ArffDatasetAdapter();
153153

154154
String mClass = Util.getClassName(instances);

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/util/Util.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -294,10 +294,10 @@ public static String getClassName(Instances instances) {
294294
*
295295
* @param instances WEKA instances to be exported
296296
*/
297-
public static String exportInstancesToArff(Instances instances) {
297+
public static String exportInstancesToArff(Instances instances, String source) {
298298
ArffSaver saver = new ArffSaver();
299299

300-
if (SwanPipeline.options.isExportArffData() && !SwanPipeline.options.getOutputDir().isEmpty() ) {
300+
if (SwanPipeline.options.isExportArffData() && !SwanPipeline.options.getOutputDir().isEmpty()) {
301301
// Save arff data.
302302
saver.setInstances(instances);
303303

@@ -308,16 +308,15 @@ public static String exportInstancesToArff(Instances instances) {
308308
if (instances.relationName().contains(":"))
309309
relationName = relationName.substring(0, instances.relationName().indexOf(":"));
310310

311-
String arffFile = SwanPipeline.options.getOutputDir() + File.separator + "arff-data" + File.separator + relationName + ".arff";
311+
String arffFile = SwanPipeline.options.getOutputDir() + File.separator + "arff-data" + File.separator + relationName + "-" + source + ".arff";
312312
saver.setFile(new File(arffFile));
313313
saver.writeBatch();
314314
} catch (IOException e) {
315315
e.printStackTrace();
316316
}
317317
return saver.retrieveFile().getAbsolutePath();
318318
}
319-
320-
return null;
319+
return null;
321320
}
322321

323322
/**

0 commit comments

Comments
 (0)