treeboost 0.1.0

High-performance Gradient Boosted Decision Tree engine for large-scale tabular data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

//! Python bindings for dataset module
//!
//! Provides access to:
//! - Data types: FeatureType, FeatureInfo, BinnedDataset
//! - Data loading: DatasetLoader
//! - Data splitting: HoldoutSplit, KFoldSplit
//! - Data pipeline: DataPipeline, PipelineConfig

mod loader;
mod pipeline;
mod splits;
pub(crate) mod types;

// Re-export types used by other python modules
pub(crate) use types::PyBinnedDataset;

use pyo3::prelude::*;

/// Register all dataset classes with the module
pub fn register(m: &Bound<'_, PyModule>) -> PyResult<()> {
    types::register(m)?;
    loader::register(m)?;
    pipeline::register(m)?;
    splits::register(m)?;
    Ok(())
}