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

…l identifiers.
adaa-polsl · Oct 5, 2024 · c2a3b1d · c2a3b1d
1 parent 84c9bbd
commit c2a3b1d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/adaa.analytics.rules/build.gradle b/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
 }

diff --git a/.../main/java/adaa/analytics/rules/logic/representation/ruleset/ContrastSurvivalRuleSet.java b/.../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");