Contrast set fix: group labels written in the log instead of numerical identifiers.

Author: agudys

adaa.analytics.rules/build.gradle

@@ -5,7 +5,7 @@ plugins {
     id 'java'
 }
 
-version = '2.1.19'
+version = '2.1.20'
 java {
     sourceCompatibility = JavaVersion.VERSION_1_8
 }

adaa.analytics.rules/src/main/java/adaa/analytics/rules/logic/representation/ruleset/ContrastSurvivalRuleSet.java

@@ -62,14 +62,17 @@ public String toString() {
         // get times from training estimator
         ArrayList<Double> times = trainingEstimator.getTimes();
 
+        ContrastSurvivalExampleSet ces = (ContrastSurvivalExampleSet)getTrainingHeader();
+        List<String> groups = ces.getContrastAttribute().getMapping().getValues();
+
         // build header
         sb.append("time,entire-set");
         for (int i = 0; i < groupEstimators.size(); ++i) {
-            sb.append(",group-" + (i + 1));
+            sb.append(",group-" + groups.get(i));
         }
 
         for (int i = 0; i < rules.size(); ++i) {
-            sb.append(",cs-" + (i + 1));
+            sb.append(",cs-" + (i+1));
         }
         sb.append("\n");