diff --git a/src/dmatrix.rs b/src/dmatrix.rs index f29b5d0..212b8b7 100644 --- a/src/dmatrix.rs +++ b/src/dmatrix.rs @@ -112,7 +112,7 @@ impl DMatrix { xgb_call!(xgboost_sys::XGDMatrixCreateFromMat(data.as_ptr(), num_rows as xgboost_sys::bst_ulong, (data.len() / num_rows) as xgboost_sys::bst_ulong, - 0.0, // TODO: can values be missing here? + f32::NAN, &mut handle))?; Ok(DMatrix::new(handle)?) }