[jvm-packages] Add the new device parameter. (#9385)

2023-07-17 18:40:39 +08:00
parent 2caceb157d
commit f4fb2be101
15 changed files with 112 additions and 47 deletions
--- a/jvm-packages/xgboost4j/src/test/java/ml/dmlc/xgboost4j/java/BoosterImplTest.java
+++ b/jvm-packages/xgboost4j/src/test/java/ml/dmlc/xgboost4j/java/BoosterImplTest.java
@@ -143,7 +143,6 @@ public class BoosterImplTest {
    booster.saveModel(temp.getAbsolutePath());

    String modelString = new String(booster.toByteArray("json"));
-    System.out.println(modelString);

    Booster bst2 = XGBoost.loadModel(temp.getAbsolutePath());
    assert (Arrays.equals(bst2.toByteArray("ubj"), booster.toByteArray("ubj")));