[package]
edition = "2021"
rust-version = "1.74"
name = "model-selection-rs"
version = "0.1.0"
build = false
exclude = [
"/.github",
"/benches/target",
]
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Cross-validation and model-selection utilities for Rust: stratified / group-aware / time-series splitting, nested CV, and learning & validation curves. Dependency-light, composes with any modeling crate."
readme = "README.md"
keywords = [
"machine-learning",
"cross-validation",
"kfold",
"stratified",
"model-selection",
]
categories = [
"science",
"algorithms",
]
license = "MIT"
repository = "https://github.com/mi7plus/model-selection-rs"
[package.metadata.docs.rs]
all-features = true
[features]
default = []
parallel = ["dep:rayon"]
smartcore-metrics = ["dep:smartcore"]
[lib]
name = "model_selection_rs"
path = "src/lib.rs"
[[example]]
name = "cross_validate"
path = "examples/cross_validate.rs"
[[example]]
name = "kfold_family"
path = "examples/kfold_family.rs"
[[example]]
name = "learning_curve"
path = "examples/learning_curve.rs"
[[example]]
name = "nested_cv"
path = "examples/nested_cv.rs"
[[example]]
name = "shuffle_split_family"
path = "examples/shuffle_split_family.rs"
[[example]]
name = "time_series_split"
path = "examples/time_series_split.rs"
[[example]]
name = "validation_curve"
path = "examples/validation_curve.rs"
[[test]]
name = "nested_cv_bias"
path = "tests/nested_cv_bias.rs"
[[test]]
name = "splitters_consistency"
path = "tests/splitters_consistency.rs"
[[bench]]
name = "cross_validate"
path = "benches/cross_validate.rs"
harness = false
[dependencies.ndarray]
version = "0.16"
[dependencies.rand]
version = "0.8"
[dependencies.rayon]
version = "1.7"
optional = true
[dependencies.smartcore]
version = "0.3"
optional = true
[dependencies.thiserror]
version = "1"
[dev-dependencies.approx]
version = "0.5"
[dev-dependencies.criterion]
version = "0.5"