18 lines
719 B
Plaintext
18 lines
719 B
Plaintext
#include "helpers.h"
|
|
#include "../../src/data/device_adapter.cuh"
|
|
#include "../../src/data/device_dmatrix.h"
|
|
|
|
namespace xgboost {
|
|
std::shared_ptr<DMatrix> RandomDataGenerator::GenerateDeviceDMatrix(bool with_label,
|
|
bool float_label,
|
|
size_t classes) {
|
|
std::vector<HostDeviceVector<float>> storage(cols_);
|
|
std::string arr = this->GenerateColumnarArrayInterface(&storage);
|
|
auto adapter = data::CudfAdapter(arr);
|
|
std::shared_ptr<DMatrix> m {
|
|
new data::DeviceDMatrix{&adapter,
|
|
std::numeric_limits<float>::quiet_NaN(), 1, 256}};
|
|
return m;
|
|
}
|
|
} // namespace xgboost
|