Merge pull request #20 from vbkaisetsu/patch-2

davechallis · web-flow · commit 296cf9ea6abd · 2024-02-02T13:05:18.000Z
Change missing from 0 to NaN
diff --git 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)?)
     }

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ impl DMatrix {`
`112`	`112`	`xgb_call!(xgboost_sys::XGDMatrixCreateFromMat(data.as_ptr(),`
`113`	`113`	`num_rows as xgboost_sys::bst_ulong,`
`114`	`114`	`(data.len() / num_rows) as xgboost_sys::bst_ulong,`
`115`		`- 0.0, // TODO: can values be missing here?`
	`115`	`+ f32::NAN,`
`116`	`116`	`&mut handle))?;`
`117`	`117`	`Ok(DMatrix::new(handle)?)`
`118`	`118`	`}`