[jvm-packages] support missing value when constructing dmatrix with i…

…terator (#10628)
dmlc · Jul 23, 2024 · 7949a8d · 7949a8d
1 parent b3ed818
commit 7949a8d
Show file tree

Hide file tree

Showing 8 changed files with 294 additions and 156 deletions.
diff --git a/include/xgboost/c_api.h b/include/xgboost/c_api.h
@@ -402,13 +402,15 @@ XGB_EXTERN_C typedef int XGBCallbackDataIterNext(  // NOLINT(*)
  * \param data_handle The handle to the data.
  * \param callback The callback to get the data.
  * \param cache_info Additional information about cache file, can be null.
+ * \param missing Which value to represent missing value.
  * \param out The created DMatrix
  * \return 0 when success, -1 when failure happens.
  */
 XGB_DLL int XGDMatrixCreateFromDataIter(
     DataIterHandle data_handle,
     XGBCallbackDataIterNext* callback,
     const char* cache_info,
+    float missing,
     DMatrixHandle *out);
 
 /**