Refer to predicted value from Counterfactual data

Signed-off-by: Gaurav Gupta <[email protected]>
microsoft · Oct 30, 2023 · a21327a · a21327a
1 parent c0e56ed
commit a21327a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/libs/counterfactuals/src/util/getOriginalData.ts b/libs/counterfactuals/src/util/getOriginalData.ts
@@ -4,7 +4,6 @@
 import {
   ICounterfactualData,
   IDataset,
-  ifEnableLargeData,
   JointDataset
 } from "@responsible-ai/core-ui";
 import { localization } from "@responsible-ai/localization";
@@ -21,7 +20,7 @@ export function getOriginalData(
       index
     )
   };
-  if (ifEnableLargeData(dataset) && counterfactualData) {
+  if (counterfactualData) {
     const featureNames = counterfactualData.feature_names_including_target;
     const dataPoint = counterfactualData.test_data[0][0];
     featureNames.forEach((f, index) => {