Counterfactual Chart: Correct target description according to task_ty…

…pe (#1261) * Counterfactual Chart: Correct target description according to task_type Signed-off-by: Gaurav Gupta <[email protected]> * Change function name Signed-off-by: Gaurav Gupta <[email protected]> * Make Range lowercase Signed-off-by: Gaurav Gupta <[email protected]>
microsoft · Mar 8, 2022 · 1f58ab0 · 1f58ab0
1 parent cf6f2f5
commit 1f58ab0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/libs/counterfactuals/src/lib/CounterfactualChart.tsx b/libs/counterfactuals/src/lib/CounterfactualChart.tsx
@@ -327,7 +327,7 @@ export class CounterfactualChart extends React.PureComponent<
                   styles={FabricStyles.smallDropdownStyle}
                 />
                 <div className={classNames.legendLabel}>
-                  <b>{`${localization.Counterfactuals.currentClass}: `}</b>
+                  <b>{`${this.getTargetDescription()}: `}</b>
                   {this.getCurrentLabel()}
                 </div>
                 <PrimaryButton
@@ -391,6 +391,13 @@ export class CounterfactualChart extends React.PureComponent<
     return this.props.data.desired_class || "";
   }
 
+  private getTargetDescription(): string {
+    if (this.context.dataset.task_type === "regression") {
+      return localization.Counterfactuals.currentRange;
+    }
+    return localization.Counterfactuals.currentClass;
+  }
+
   private buildRowOptions(cohortIndex: number): void {
     this.context.errorCohorts[cohortIndex].cohort.sort(JointDataset.IndexLabel);
   }

diff --git a/libs/localization/src/lib/en.json b/libs/localization/src/lib/en.json
@@ -109,6 +109,7 @@
     "createCounterfactual": "Counterfactual",
     "createOwn": "Create your own counterfactual:",
     "currentClass": "Current class",
+    "currentRange": "Current range",
     "decreaseByOne": "Decrease value by 1",
     "desiredClass": "Desired class for counterfactual(s)",
     "header": "What-If counterfactuals",