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

[XGBoost4J-Spark] bestIteration and bestScore for early stopping (#7095)

Browse Source
This commit is contained in:
naveenkb
2021-07-19 16:16:49 +05:30
committed by GitHub
parent d7c14496d2
commit 9f7f8b976d
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j/src/test/java/ml/dmlc/xgboost4j/java/BoosterImplTest.java
View File

@@ -643,7 +643,7 @@ public class BoosterImplTest {
}});
Map<String, String> attr = booster.getAttrs();
TestCase.assertEquals(attr.size(), 4);
TestCase.assertEquals(attr.size(), 6);
TestCase.assertEquals(attr.get("testKey1"), "testValue2");
TestCase.assertEquals(attr.get("aa"), "AA");
TestCase.assertEquals(attr.get("bb"), "BB");