Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Protected Attributes
xgboost::ArrayIterForTest Class Referenceabstract
Inheritance diagram for xgboost::ArrayIterForTest:
xgboost::CudaArrayIterForTest xgboost::NumpyArrayIterForTest

Public Member Functions

std::string AsArray () const
 
virtual int Next ()=0
 
virtual void Reset ()
 
std::size_t Iter () const
 
auto Proxy () -> decltype(proxy_)
 
 ArrayIterForTest (float sparsity, size_t rows, size_t cols, size_t batches)
 
 ArrayIterForTest (Context const *ctx, HostDeviceVector< float > const &data, std::size_t n_samples, bst_feature_t n_features, std::size_t n_batches)
 Create iterator with user provided data.
 

Static Public Member Functions

static size_t constexpr Rows ()
 
static size_t constexpr Batches ()
 
static size_t constexpr Cols ()
 

Protected Attributes

HostDeviceVector< float > data_
 
size_t iter_ {0}
 
DMatrixHandle proxy_
 
std::unique_ptr< RandomDataGeneratorrng_
 
std::vector< std::string > batches_
 
std::string interface_
 
size_t rows_
 
size_t cols_
 
size_t n_batches_
 

The documentation for this class was generated from the following files: