treeboost 0.1.0

[[bin]]
name = "treeboost"
path = "src/main.rs"

[dependencies.bincode]
version = "1.3"

[dependencies.bytemuck]
features = ["derive"]
version = "1.19"

[dependencies.chrono]
version = "0.4"

[dependencies.clap]
features = ["derive"]
version = "4.5"

[dependencies.crc32fast]
version = "1.5"

[dependencies.csv]
version = "1.3"

[dependencies.cudarc]
default-features = false
features = ["driver", "nvrtc", "dynamic-linking", "cuda-version-from-build-system"]
optional = true
version = "=0.18.2"

[dependencies.faer]
version = "0.23"

[dependencies.fs4]
version = "0.13"

[dependencies.memmap2]
optional = true
version = "0.9"

[dependencies.numpy]
optional = true
version = "0.23"

[dependencies.ordered-float]
version = "4.5"

[dependencies.polars]
features = ["parquet", "lazy", "dtype-full"]
version = "0.51"

[dependencies.pollster]
optional = true
version = "0.4"

[dependencies.pyo3]
features = ["extension-module"]
optional = true
version = "0.23"

[dependencies.rand]
version = "0.8"

[dependencies.rayon]
version = "1.10"

[dependencies.rkyv]
version = "0.8"

[dependencies.rustc-hash]
version = "2.1"

[dependencies.serde]
features = ["derive"]
version = "1.0"

[dependencies.serde_json]
version = "1.0"

[dependencies.tdigest]
version = "0.2"

[dependencies.thiserror]
version = "2.0"

[dependencies.wgpu]
optional = true
version = "27"

[dev-dependencies.approx]
version = "0.5"

[dev-dependencies.criterion]
features = ["html_reports"]
version = "0.5"

[dev-dependencies.fastrand]
version = "2.3"

[dev-dependencies.forust-ml]
version = "0.4"

[dev-dependencies.gbdt]
features = ["enable_training"]
version = "0.1"

[dev-dependencies.tempfile]
version = "3.15"

[[example]]
name = "autotuner"
path = "examples/autotuner.rs"

[[example]]
name = "basic_regression"
path = "examples/basic_regression.rs"

[[example]]
name = "classification"
path = "examples/classification.rs"

[[example]]
name = "conformal_prediction"
path = "examples/conformal_prediction.rs"

[[example]]
name = "linear_tree"
path = "examples/linear_tree.rs"

[[example]]
name = "public_data_download"
path = "examples/public_data_download.rs"

[[example]]
name = "target_encoding"
path = "examples/target_encoding.rs"

[[example]]
name = "universal_model"
path = "examples/universal_model.rs"

[features]
cuda = ["dep:cudarc"]
default = []
gpu = ["dep:wgpu", "dep:pollster"]
mmap = ["dep:memmap2"]
python = ["dep:pyo3", "dep:numpy"]

[lib]
crate-type = ["cdylib", "rlib"]
name = "treeboost"
path = "src/lib.rs"

[package]
authors = ["Farhan Syah"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "algorithms", "command-line-utilities"]
description = "High-performance Gradient Boosted Decision Tree engine for large-scale tabular data"
documentation = "https://docs.rs/treeboost"
edition = "2021"
exclude = ["benchmarks/**", "benches/**", "catboost_info/**", "images/**", "results/**", "samples/**", "scripts/**", "target/**"]
homepage = "https://github.com/ml-rust/treeboost"
keywords = ["machine-learning", "gbdt", "gradient-boosting", "decision-tree", "xgboost"]
license = "Apache-2.0"
name = "treeboost"
readme = "README.md"
repository = "https://github.com/ml-rust/treeboost"
version = "0.1.0"

[profile.bench]
codegen-units = 1
lto = "thin"

[profile.release]
codegen-units = 1
lto = "thin"
opt-level = 3

[[test]]
name = "automl_integration"
path = "tests/automl_integration.rs"

[[test]]
name = "automl_ltt"
path = "tests/automl_ltt.rs"

[[test]]
name = "booster"
path = "tests/booster.rs"

[[test]]
name = "encoding"
path = "tests/encoding.rs"

[[test]]
name = "features"
path = "tests/features.rs"

[[test]]
name = "gpu"
path = "tests/gpu.rs"

[[test]]
name = "monitoring"
path = "tests/monitoring.rs"

[[test]]
name = "optimization"
path = "tests/optimization.rs"

[[test]]
name = "parquet"
path = "tests/parquet.rs"

[[test]]
name = "pipeline"
path = "tests/pipeline.rs"

[[test]]
name = "preprocessing"
path = "tests/preprocessing.rs"

[[test]]
name = "progress_callback"
path = "tests/progress_callback.rs"

[[test]]
name = "serialization"
path = "tests/serialization.rs"

[[test]]
name = "tuner"
path = "tests/tuner.rs"