Skip to content
Snippets Groups Projects
Commit bac2951f authored by Steve Reis's avatar Steve Reis
Browse files

fix: enabled LR Cross-Validation, Cast numbers -> strings

parent e6eb0474
No related branches found
No related tags found
No related merge requests found
......@@ -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,10 +61,7 @@ export default class LogisticRegressionCVHandler extends BaseHandler {
name: lookupDict[key],
type: 'string',
})),
data: data.summary['row_names'].map((key: any, i: number) => {
// could be optimized
return [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]}`)])
};
}
......@@ -131,6 +128,8 @@ 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);
});
}
}
......@@ -9,7 +9,7 @@ const transformToAlgorithms = jsonata(`
$checkVal:= function($val) { $val ? $val : undefined};
$excludedParams:= ['centers', 'formula'];
$includes:= ['ANOVA_ONEWAY','ANOVA','LINEAR_REGRESSION',
'LOGISTIC_REGRESSION','TTEST_INDEPENDENT','TTEST_PAIRED',
'LOGISTIC_REGRESSION', 'LOGISTIC_REGRESSION_CV','TTEST_INDEPENDENT','TTEST_PAIRED',
'PEARSON','ID3','KMEANS','NAIVE_BAYES',
'TTEST_ONESAMPLE','PCA','CALIBRATION_BELT','CART',
'KAPLAN_MEIER','THREE_C', 'ONE_WAY_ANOVA', 'PEARSON_CORRELATION', 'LINEAR_REGRESSION_CV', 'TTEST_ONESAMPLE', 'PAIRED_TTEST'];
......
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