Struct BinnedDataset 

pub struct BinnedDataset {
    pub num_rows: usize,
    pub row_offset: u64,
    pub features: Vec<FeatureMeta>,
    pub groups: Vec<FeatureGroup>,
    pub total_bins: usize,
    pub label: Vec<f32>,
    pub weight: Option<Vec<f32>>,
    pub query_boundaries: Option<Vec<u32>>,
}

Fields

num_rows: usize

row_offset: u64

Global id of local row 0: rows of lower ranks come first. Keys the random draws of quantization and bagging.

features: Vec<FeatureMeta>

Every input feature in input order, trivial ones included.

groups: Vec<FeatureGroup>

total_bins: usize

label: Vec<f32>

weight: Option<Vec<f32>>

query_boundaries: Option<Vec<u32>>

num_queries + 1 row boundaries; queries are contiguous runs of rows.

Implementations

impl BinnedDataset

pub fn construct( input: &DatasetInput<'_>, params: &Params, collective: &mut dyn Collective, ) -> Result<Self>

Bins a training set. Bin mappers come from a sample of bin_construct_sample_cnt rows over every rank (allgathered), so every rank builds identical mappers; row_offset is the rows of lower ranks.

pub fn construct_with_reference( input: &DatasetInput<'_>, reference: &BinnedDataset, collective: &mut dyn Collective, ) -> Result<Self>

Bins a validation set with reference’s mappers and group layout; feature names and categorical declarations come from reference.

impl BinnedDataset

pub fn num_queries(&self) -> usize

pub fn bin(&self, feature: usize, row: usize) -> u8

The bin of feature at row.

Trait Implementations

impl Clone for BinnedDataset

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BinnedDataset

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for BinnedDataset

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for BinnedDataset