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

//! Python bindings for tuner module
//!
//! Provides access to:
//! - AutoTuner: Main hyperparameter tuning interface
//! - TunerConfig, ParameterSpace, ParamBounds: Configuration
//! - TuningMode, OptimizationMetric, TaskType: Enum wrappers
//! - GridStrategy, EvalStrategy, ModelFormat: Strategy enums
//! - TrialResult, SearchHistory: Result types

mod autotuner;
mod callback;
mod config;
mod enums;
mod results;

use pyo3::prelude::*;

/// Register all tuner classes with the module
pub fn register(m: &Bound<'_, PyModule>) -> PyResult<()> {
    enums::register(m)?;
    config::register(m)?;
    results::register(m)?;
    autotuner::register(m)?;
    Ok(())
}