将 xgb.DMatrix 对象保存到二进制文件 — xgb.DMatrix.save • xgboost - XGBoost 文档

将 xgb.DMatrix 对象保存到二进制文件

用法

xgb.DMatrix.save(dmatrix, fname)

参数

dmatrix: xgb.DMatrix 对象
fname: 要写入的文件名。

示例

DONTSHOW({RhpcBLASctl::omp_set_num_threads(1)})
data(agaricus.train, package = "xgboost")

dtrain <- with(agaricus.train, xgb.DMatrix(data, label = label, nthread = 2))
fname <- file.path(tempdir(), "xgb.DMatrix.data")
xgb.DMatrix.save(dtrain, fname)
dtrain <- xgb.DMatrix(fname, nthread = 1)