Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[jvm-packages] XGBoost Spark should deal with NaN when parsing evaluation output (#5546)

Browse Source
This commit is contained in:
Liang-Chi Hsieh
2020-04-19 23:10:30 -07:00
committed by GitHub
parent b809f5d8b8
commit 397d8f0ee7
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/Booster.java
View File

@@ -242,7 +242,14 @@ public class Booster implements Serializable, KryoSerializable {
String stringFormat = evalSet(evalMatrixs, evalNames, iter);
String[] metricPairs = stringFormat.split("\t");
for (int i = 1; i < metricPairs.length; i++) {
metricsOut[i - 1] = Float.valueOf(metricPairs[i].split(":")[1]);
String value = metricPairs[i].split(":")[1];
if (value.equalsIgnoreCase("nan")) {
metricsOut[i - 1] = Float.NaN;
} else if (value.equalsIgnoreCase("-nan")) {
metricsOut[i - 1] = -Float.NaN;
} else {
metricsOut[i - 1] = Float.valueOf(value);
}
}
return stringFormat;
}