Contrast set fix: group labels written in the log instead of numerica…

…l identifiers.
adaa-polsl · Oct 4, 2024 · d8a0ed1 · d8a0ed1
1 parent 8ffe07f
commit d8a0ed1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/adaa.analytics.rules/build.gradle b/adaa.analytics.rules/build.gradle
@@ -27,7 +27,7 @@ codeQuality {
 }
 
 sourceCompatibility = 1.8
-version = '1.7.20'
+version = '1.7.21'
 
 
 jar {

diff --git a/...ules/src/main/java/adaa/analytics/rules/logic/representation/ContrastSurvivalRuleSet.java b/...ules/src/main/java/adaa/analytics/rules/logic/representation/ContrastSurvivalRuleSet.java
@@ -62,13 +62,16 @@ public String toString() {
         sb.append(super.toString());
         sb.append("\nEstimator:\n");
 
+        IContrastExampleSet ces = (IContrastExampleSet)trainingSet;
+        List<String> groups = ces.getContrastAttribute().getMapping().getValues();
+
         // get times from training estimator
         ArrayList<Double> times = trainingEstimator.getTimes();
 
         // 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) {