honeymoose/OpenSearch
1
0
Fork 0
mirror of https://github.com/honeymoose/OpenSearch.git synced 2025-02-25 06:16:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ML] Fix binary classification importance in LegacyFeatureImportanceTests (#63000)

Browse Source 
Fixes #62991
This commit is contained in:
Dimitris Athanasiou 2020-09-29 15:53:34 +03:00 committed by GitHub
parent 2b9032a07d
commit facf9ede0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
x-pack/plugin/core/src/test/java/org/elasticsearch/xpack/core/ml/inference/results/LegacyFeatureImportanceTests.java
View File

@ -29,7 +29,8 @@ public class LegacyFeatureImportanceTests extends AbstractWireSerializingTestCas
.map(featureName -> new LegacyFeatureImportance.ClassImportance(featureName, randomDouble()))
.collect(Collectors.toList());
importance = classImportances.stream().mapToDouble(LegacyFeatureImportance.ClassImportance::getImportance).map(Math::abs).sum();
importance = classImportances.size() == 2 ? Math.abs(classImportances.get(0).getImportance()) :
classImportances.stream().mapToDouble(LegacyFeatureImportance.ClassImportance::getImportance).map(Math::abs).sum();
}
return new LegacyFeatureImportance(randomAlphaOfLength(10), importance, classImportances);
}