[package]
edition = "2024"
rust-version = "1.93"
name = "hessboost"
version = "0.2.1"
authors = ["Brenden Matthews <brenden@brndn.io>"]
build = false
include = [
"/src/**",
"/tests/**",
"/examples/**",
"/benches/**",
"/README.md",
"/LICENSE",
]
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Fast, deterministic gradient boosting (GBDT) in Rust: conformal intervals, explainable boosting machines, distributional boosting, tree-based diffusion, and XGBoost model interchange"
readme = "README.md"
keywords = [
"xgboost",
"gradient-boosting",
"gbdt",
"machine-learning",
"decision-trees",
]
categories = [
"artificial-intelligence",
"algorithms",
"science",
]
license = "Apache-2.0"
repository = "https://github.com/brndnmtthws/hessboost"
[features]
metal = [
"dep:objc2",
"dep:objc2-metal",
"dep:objc2-foundation",
]
[lib]
name = "hessboost"
path = "src/lib.rs"
[[example]]
name = "balanced_bagging"
path = "examples/balanced_bagging.rs"
[[example]]
name = "bench_compare"
path = "examples/bench_compare.rs"
[[example]]
name = "binary_classification"
path = "examples/binary_classification.rs"
[[example]]
name = "boulevard_inference"
path = "examples/boulevard_inference.rs"
[[example]]
name = "budget"
path = "examples/budget.rs"
[[example]]
name = "compact_model"
path = "examples/compact_model.rs"
[[example]]
name = "conformal"
path = "examples/conformal.rs"
[[example]]
name = "constraints"
path = "examples/constraints.rs"
[[example]]
name = "custom_objective"
path = "examples/custom_objective.rs"
[[example]]
name = "distributional"
path = "examples/distributional.rs"
[[example]]
name = "ebm"
path = "examples/ebm.rs"
[[example]]
name = "forest_flow"
path = "examples/forest_flow.rs"
[[example]]
name = "metal"
path = "examples/metal.rs"
[[example]]
name = "model_io"
path = "examples/model_io.rs"
[[example]]
name = "multiclass"
path = "examples/multiclass.rs"
[[example]]
name = "online_update"
path = "examples/online_update.rs"
[[example]]
name = "ordered_target_stats"
path = "examples/ordered_target_stats.rs"
[[example]]
name = "pfn_boost"
path = "examples/pfn_boost.rs"
[[example]]
name = "rank_xendcg"
path = "examples/rank_xendcg.rs"
[[example]]
name = "ranking"
path = "examples/ranking.rs"
[[example]]
name = "shap"
path = "examples/shap.rs"
[[example]]
name = "train_regression"
path = "examples/train_regression.rs"
[[example]]
name = "tree_diffusion"
path = "examples/tree_diffusion.rs"
[[example]]
name = "virtual_ensembles"
path = "examples/virtual_ensembles.rs"
[[test]]
name = "boulevard"
path = "tests/boulevard.rs"
[[test]]
name = "budget"
path = "tests/budget.rs"
[[test]]
name = "continuation"
path = "tests/continuation.rs"
[[test]]
name = "diffusion"
path = "tests/diffusion.rs"
[[test]]
name = "distributional"
path = "tests/distributional.rs"
[[test]]
name = "ebm"
path = "tests/ebm.rs"
[[test]]
name = "forest"
path = "tests/forest.rs"
[[test]]
name = "lightgbm_parity"
path = "tests/lightgbm_parity.rs"
[[test]]
name = "metal"
path = "tests/metal.rs"
[[test]]
name = "model_format"
path = "tests/model_format.rs"
[[test]]
name = "multi_output"
path = "tests/multi_output.rs"
[[test]]
name = "native_format"
path = "tests/native_format.rs"
[[test]]
name = "online"
path = "tests/online.rs"
[[test]]
name = "parity"
path = "tests/parity.rs"
[[test]]
name = "properties"
path = "tests/properties.rs"
[[test]]
name = "quantized"
path = "tests/quantized.rs"
[[test]]
name = "round_hook"
path = "tests/round_hook.rs"
[[test]]
name = "sampling"
path = "tests/sampling.rs"
[[test]]
name = "sglb"
path = "tests/sglb.rs"
[[test]]
name = "shap_accumulation"
path = "tests/shap_accumulation.rs"
[[test]]
name = "target_stats"
path = "tests/target_stats.rs"
[[test]]
name = "tree_options"
path = "tests/tree_options.rs"
[[bench]]
name = "training"
path = "benches/training.rs"
harness = false
[dependencies.rayon]
version = "1.12"
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0"
features = ["float_roundtrip"]
[dependencies.zstd]
version = "0.14"
default-features = false
[dev-dependencies.approx]
version = "0.5.1"
[dev-dependencies.criterion]
version = "0.8.2"
[dev-dependencies.proptest]
version = "1.11"
[dev-dependencies.rand]
version = "0.10.3"
[target.'cfg(target_os = "macos")'.dependencies.objc2]
version = "0.6"
features = ["std"]
optional = true
default-features = false
[target.'cfg(target_os = "macos")'.dependencies.objc2-foundation]
version = "0.3"
features = [
"std",
"NSString",
]
optional = true
default-features = false
[target.'cfg(target_os = "macos")'.dependencies.objc2-metal]
version = "0.3"
features = [
"std",
"MTLDevice",
"MTLCommandQueue",
"MTLCommandBuffer",
"MTLComputeCommandEncoder",
"MTLCommandEncoder",
"MTLLibrary",
"MTLComputePipeline",
"MTLAllocation",
"MTLResource",
"MTLBuffer",
"MTLTypes",
]
optional = true
default-features = false
[lints.clippy]
cast_possible_truncation = "allow"
cast_possible_wrap = "allow"
cast_precision_loss = "allow"
cast_sign_loss = "allow"
float_cmp = "allow"
inline_always = "allow"
many_single_char_names = "allow"
missing_errors_doc = "allow"
missing_panics_doc = "allow"
must_use_candidate = "allow"
similar_names = "allow"
too_many_lines = "allow"
undocumented_unsafe_blocks = "warn"
[lints.clippy.pedantic]
level = "warn"
priority = -1
[profile.bench]
opt-level = 3
lto = "thin"
codegen-units = 1
[profile.ci]
opt-level = 1
debug = "line-tables-only"
inherits = "dev"
[profile.release]
opt-level = 3
lto = "thin"
codegen-units = 1
panic = "abort"