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

[jvm-packages] Deprecate constructors with implicit missing value. (#7225)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-09-17 04:35:04 +08:00
committed by GitHub
parent 0ed979b096
commit 9f63d6fead
4 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
jvm-packages/xgboost4j/src/test/java/ml/dmlc/xgboost4j/java/DMatrixTest.java
View File

@@ -212,7 +212,7 @@ public class DMatrixTest {
label0[i] = random.nextFloat();
}
DMatrix dmat0 = new DMatrix(data0, nrow, ncol);
DMatrix dmat0 = new DMatrix(data0, nrow, ncol, Float.NaN);
dmat0.setLabel(label0);
//check
@@ -281,7 +281,7 @@ public class DMatrixTest {
label0[i] = random.nextFloat();
}
dmat0 = new DMatrix(data0);
dmat0 = new DMatrix(data0, Float.NaN);
dmat0.setLabel(label0);
//check
@@ -318,7 +318,7 @@ public class DMatrixTest {
for (int j = 0; j < data0.ncol; j++)
data0.set(i, j, data[i][j]);
trainMat = new DMatrix(data0);
trainMat = new DMatrix(data0, Float.NaN);
trainMat.setLabel(new float[]{1f, 2f, 3f});
HashMap<String, Object> params = new HashMap<>();
@@ -338,7 +338,7 @@ public class DMatrixTest {
// (3,1) -> 2
// (2,3) -> 3
for (int i = 0; i < 3; i++) {
float[][] preds = booster.predict(new DMatrix(data[i], 1, 2));
float[][] preds = booster.predict(new DMatrix(data[i], 1, 2, Float.NaN));
assertEquals(1, preds.length);
assertArrayEquals(new float[]{(float) (i + 1)}, preds[0], 1e-2f);
}

2
jvm-packages/xgboost4j/src/test/scala/ml/dmlc/xgboost4j/scala/DMatrixSuite.scala
View File

@@ -130,7 +130,7 @@ class DMatrixSuite extends FunSuite {
for (i <- label0.indices) {
label0(i) = Random.nextFloat()
}
val dmat0 = new DMatrix(data0, nrow, ncol)
val dmat0 = new DMatrix(data0, nrow, ncol, Float.NaN)
dmat0.setLabel(label0)
// check
assert(dmat0.rowNum === 10)