From a51616d34830a410fc01baae51349435c1777764 Mon Sep 17 00:00:00 2001
From: stevereis <stevereis93@gmail.com>
Date: Sat, 11 Mar 2023 17:50:03 +0100
Subject: [PATCH] fix(issue-71): Logitistic regression cv matrix
---
.../logistic-regression-cv.handler.ts | 20 +++++++++++++------
1 file changed, 14 insertions(+), 6 deletions(-)
diff --git a/api/src/engine/connectors/exareme/handlers/algorithms/logistic-regression-cv.handler.ts b/api/src/engine/connectors/exareme/handlers/algorithms/logistic-regression-cv.handler.ts
index cded114..588bcca 100644
--- a/api/src/engine/connectors/exareme/handlers/algorithms/logistic-regression-cv.handler.ts
+++ b/api/src/engine/connectors/exareme/handlers/algorithms/logistic-regression-cv.handler.ts
@@ -47,7 +47,7 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
private canHandle(experiment: Experiment, data: unknown): boolean {
return (
experiment.algorithm.name.toLowerCase() ===
- LogisticRegressionCVHandler.ALGO_NAME &&
+ LogisticRegressionCVHandler.ALGO_NAME &&
!!data &&
!!data[0] &&
!!data[0]['summary']
@@ -61,7 +61,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
name: lookupDict[key],
type: 'string',
})),
- data: data.summary['row_names'].map((key: any, i: number) => [key, ...keys.map((k) => `${data['summary'][k][i]}`)])
+ data: data.summary['row_names'].map((key: any, i: number) => [
+ key,
+ ...keys.map((k) => `${data['summary'][k][i]}`),
+ ]),
};
}
@@ -70,7 +73,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
return {
name: 'Confusion matrix',
- matrix: [[matrix['tp']], [matrix['fp']], [matrix['fn']], [matrix['tn']]],
+ matrix: [
+ [matrix['tp'], matrix['fp']],
+ [matrix['fn'], matrix['tn']],
+ ],
xAxis: {
categories: ['Positive', 'Negative'],
label: 'Actual Values',
@@ -128,8 +134,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
this.getROC(improvedData),
];
- results.filter((r) => !!r).forEach((r) => {
- experiment.results.push(r);
- });
+ results
+ .filter((r) => !!r)
+ .forEach((r) => {
+ experiment.results.push(r);
+ });
}
}
--
GitLab