use of org.kie.kogito.explainability.model.Saliency in project kogito-apps by kiegroup.
the class ShapKernelExplainerTest method testParallel.
@Test
void testParallel() throws InterruptedException, ExecutionException {
// establish background data and desired data to explain
double[][] largeBackground = new double[100][10];
for (int i = 0; i < 100; i++) {
for (int j = 0; j < 10; j++) {
largeBackground[i][j] = i / 100. + j;
}
}
double[][] toExplainLargeBackground = { { 0, 1., -2., 3.5, -4.1, 5.5, -12., .8, .11, 15. } };
double[][][] expected = { { { -0.495, 0., -4.495, 0.005, -8.595, 0.005, -18.495, -6.695, -8.385, 5.505 } } };
List<PredictionInput> background = createPIFromMatrix(largeBackground);
List<PredictionInput> toExplain = createPIFromMatrix(toExplainLargeBackground);
PredictionProvider model = TestUtils.getSumSkipModel(1);
ShapConfig skConfig = testConfig.withBackground(background).build();
// initialize explainer
List<PredictionOutput> predictionOutputs = model.predictAsync(toExplain).get();
List<Prediction> predictions = new ArrayList<>();
for (int i = 0; i < predictionOutputs.size(); i++) {
predictions.add(new SimplePrediction(toExplain.get(i), predictionOutputs.get(i)));
}
// evaluate if the explanations match the expected value
ShapKernelExplainer ske = new ShapKernelExplainer(skConfig);
CompletableFuture<ShapResults> explanationsCF = ske.explainAsync(predictions.get(0), model);
ExecutorService executor = ForkJoinPool.commonPool();
executor.submit(() -> {
Saliency[] explanationSaliencies = explanationsCF.join().getSaliencies();
RealMatrix explanations = saliencyToMatrix(explanationSaliencies)[0];
assertArrayEquals(expected[0][0], explanations.getRow(0), 1e-2);
});
}
use of org.kie.kogito.explainability.model.Saliency in project kogito-apps by kiegroup.
the class ShapKernelExplainerTest method testLargeBackground2.
@Test
void testLargeBackground2() throws InterruptedException, TimeoutException, ExecutionException {
// establish background data and desired data to explain
double[][] largeBackground = new double[100][10];
for (int i = 0; i < 100; i++) {
for (int j = 0; j < 10; j++) {
largeBackground[i][j] = i / 100. + j;
}
}
double[][] toExplainLargeBackground = { { 0, 1., -2., 3.5, -4.1, 5.5, -12., .8, .11, 15. } };
double[][][] expected = { { { -0.495, 0., -4.495, 0.005, -8.595, 0.005, -18.495, -6.695, -8.385, 5.505 } } };
List<PredictionInput> background = createPIFromMatrix(largeBackground);
List<PredictionInput> toExplain = createPIFromMatrix(toExplainLargeBackground);
PredictionProvider model = TestUtils.getSumSkipModel(1);
ShapConfig skConfig = testConfig.withBackground(background).withNSamples(1000).build();
// initialize explainer
List<PredictionOutput> predictionOutputs = model.predictAsync(toExplain).get();
List<Prediction> predictions = new ArrayList<>();
for (int i = 0; i < predictionOutputs.size(); i++) {
predictions.add(new SimplePrediction(toExplain.get(i), predictionOutputs.get(i)));
}
// evaluate if the explanations match the expected value
ShapKernelExplainer ske = new ShapKernelExplainer(skConfig);
for (int i = 0; i < toExplain.size(); i++) {
Saliency[] explanationSaliencies = ske.explainAsync(predictions.get(i), model).get(5, TimeUnit.SECONDS).getSaliencies();
RealMatrix[] explanationsAndConfs = saliencyToMatrix(explanationSaliencies);
RealMatrix explanations = explanationsAndConfs[0];
for (int j = 0; j < explanations.getRowDimension(); j++) {
assertArrayEquals(expected[i][j], explanations.getRow(j), 1e-2);
}
}
}
use of org.kie.kogito.explainability.model.Saliency in project kogito-apps by kiegroup.
the class ShapKernelExplainerTest method testRegularizations.
@ParameterizedTest
@ValueSource(ints = { 0, 1, 2, 3 })
void testRegularizations(int config) throws InterruptedException, ExecutionException {
PredictionProvider model = TestUtils.getSumSkipModel(1);
List<PredictionInput> toExplain = createPIFromMatrix(toExplainRegTests);
RealMatrix toExplainMatrix = MatrixUtils.createRealMatrix(toExplainRegTests);
List<PredictionOutput> predictionOutputs = model.predictAsync(toExplain).get();
RealVector predictionOutputVector = MatrixUtilsExtensions.vectorFromPredictionOutput(predictionOutputs.get(0));
Prediction p = new SimplePrediction(toExplain.get(0), predictionOutputs.get(0));
ShapConfig skConfig = sks.get(config);
ShapKernelExplainer ske = new ShapKernelExplainer(skConfig);
ShapResults shapResults = ske.explainAsync(p, model).get();
Saliency[] saliencies = shapResults.getSaliencies();
RealMatrix[] explanationsAndConfs = saliencyToMatrix(saliencies);
RealMatrix explanations = explanationsAndConfs[0];
double actualOut = predictionOutputVector.getEntry(0);
double predOut = MatrixUtilsExtensions.sum(explanations.getRowVector(0)) + shapResults.getFnull().getEntry(0);
assertTrue(Math.abs(predOut - actualOut) < 1e-6);
}
use of org.kie.kogito.explainability.model.Saliency in project kogito-apps by kiegroup.
the class ShapKernelExplainerTest method testErrorBounds.
// given a noisy model, expect the n% confidence window to include true value roughly n% of the time
@ParameterizedTest
@ValueSource(doubles = { .001, .1, .25, .5 })
void testErrorBounds(double noise) throws InterruptedException, ExecutionException {
for (double interval : new double[] { .95, .975, .99 }) {
int[] testResults = new int[600];
for (int test = 0; test < 100; test++) {
PredictionProvider model = TestUtils.getNoisySumModel(pc.getRandom(), noise);
ShapConfig skConfig = testConfig.withBackground(createPIFromMatrix(backgroundAllZeros)).withConfidence(interval).build();
List<PredictionInput> toExplain = createPIFromMatrix(toExplainAllOnes);
ShapKernelExplainer ske = new ShapKernelExplainer(skConfig);
List<PredictionOutput> predictionOutputs = model.predictAsync(toExplain).get();
Prediction p = new SimplePrediction(toExplain.get(0), predictionOutputs.get(0));
Saliency[] saliencies = ske.explainAsync(p, model).get().getSaliencies();
RealMatrix[] explanationsAndConfs = saliencyToMatrix(saliencies);
RealMatrix explanations = explanationsAndConfs[0];
RealMatrix confidence = explanationsAndConfs[1];
for (int i = 0; i < explanations.getRowDimension(); i++) {
for (int j = 0; j < explanations.getColumnDimension(); j++) {
double conf = confidence.getEntry(i, j);
double exp = explanations.getEntry(i, j);
// see if true value falls into confidence interval
testResults[test * 6 + j] = (exp + conf) > 1.0 & 1.0 > (exp - conf) ? 1 : 0;
}
}
}
// roughly interval% of the tests should be true
double score = Arrays.stream(testResults).sum() / 600.;
assertEquals(interval, score, .05);
}
}
use of org.kie.kogito.explainability.model.Saliency in project kogito-apps by kiegroup.
the class ShapKernelExplainerTest method testManyFeatureRegularization2.
@Test
void testManyFeatureRegularization2() throws ExecutionException, InterruptedException {
RealVector modelWeights = MatrixUtils.createRealMatrix(generateN(1, 25, "5021")).getRowVector(0);
PredictionProvider model = TestUtils.getLinearModel(modelWeights.toArray());
RealMatrix data = MatrixUtils.createRealMatrix(generateN(101, 25, "8629"));
List<PredictionInput> toExplain = createPIFromMatrix(data.getRowMatrix(100).getData());
List<PredictionOutput> predictionOutputs = model.predictAsync(toExplain).get();
RealVector predictionOutputVector = MatrixUtilsExtensions.vectorFromPredictionOutput(predictionOutputs.get(0));
Prediction p = new SimplePrediction(toExplain.get(0), predictionOutputs.get(0));
List<PredictionInput> bg = createPIFromMatrix(new double[100][25]);
ShapConfig sk = testConfig.copy().withBackground(bg).withRegularizer(ShapConfig.RegularizerType.AIC).withNSamples(5000).build();
ShapKernelExplainer ske = new ShapKernelExplainer(sk);
ShapResults shapResults = ske.explainAsync(p, model).get();
Saliency[] saliencies = shapResults.getSaliencies();
RealMatrix[] explanationsAndConfs = saliencyToMatrix(saliencies);
RealMatrix explanations = explanationsAndConfs[0];
double actualOut = predictionOutputVector.getEntry(0);
double predOut = MatrixUtilsExtensions.sum(explanations.getRowVector(0)) + shapResults.getFnull().getEntry(0);
assertTrue(Math.abs(predOut - actualOut) < 1e-6);
double coefMSE = (data.getRowVector(100).ebeMultiply(modelWeights)).getDistance(explanations.getRowVector(0));
assertTrue(coefMSE < .01);
}
Aggregations