Skip to content

Commit

Permalink
[ML] Fix type for importance
Browse files Browse the repository at this point in the history
  • Loading branch information
qn895 committed Feb 11, 2021
1 parent 17a164d commit 135d850
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,7 @@ export const getFeatureImportance = (
const featureImportance: Array<{
feature_name: string[];
classes?: Array<{ class_name: FeatureImportanceClassName[]; importance: number[] }>;
importance?: number;
importance?: number | number[];
}> = row[`${mlResultsField}.feature_importance`];
if (featureImportance === undefined) return [];

Expand Down

0 comments on commit 135d850

Please sign in to comment.