Skip to content
Snippets Groups Projects
Commit a51616d3 authored by stevereis's avatar stevereis
Browse files

fix(issue-71): Logitistic regression cv matrix

parent 7a5f4c6e
No related branches found
No related tags found
No related merge requests found
...@@ -47,7 +47,7 @@ export default class LogisticRegressionCVHandler extends BaseHandler { ...@@ -47,7 +47,7 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
private canHandle(experiment: Experiment, data: unknown): boolean { private canHandle(experiment: Experiment, data: unknown): boolean {
return ( return (
experiment.algorithm.name.toLowerCase() === experiment.algorithm.name.toLowerCase() ===
LogisticRegressionCVHandler.ALGO_NAME && LogisticRegressionCVHandler.ALGO_NAME &&
!!data && !!data &&
!!data[0] && !!data[0] &&
!!data[0]['summary'] !!data[0]['summary']
...@@ -61,7 +61,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler { ...@@ -61,7 +61,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
name: lookupDict[key], name: lookupDict[key],
type: 'string', 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 { ...@@ -70,7 +73,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
return { return {
name: 'Confusion matrix', name: 'Confusion matrix',
matrix: [[matrix['tp']], [matrix['fp']], [matrix['fn']], [matrix['tn']]], matrix: [
[matrix['tp'], matrix['fp']],
[matrix['fn'], matrix['tn']],
],
xAxis: { xAxis: {
categories: ['Positive', 'Negative'], categories: ['Positive', 'Negative'],
label: 'Actual Values', label: 'Actual Values',
...@@ -128,8 +134,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler { ...@@ -128,8 +134,10 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
this.getROC(improvedData), this.getROC(improvedData),
]; ];
results.filter((r) => !!r).forEach((r) => { results
experiment.results.push(r); .filter((r) => !!r)
}); .forEach((r) => {
experiment.results.push(r);
});
} }
} }
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment