model-selection-rs 0.1.0

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.
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
27
28
29
30
31
32
33
34
35
36
37
38

[package]
name = "model-selection-rs"
version = "0.1.0"
edition = "2021"
rust-version = "1.74"
license = "MIT"
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."
repository = "https://github.com/mi7plus/model-selection-rs"
readme = "README.md"
keywords = ["machine-learning", "cross-validation", "kfold", "stratified", "model-selection"]
categories = ["science", "algorithms"]
exclude = ["/.github", "/benches/target"]

[dependencies]
ndarray = "0.16"
rand = "0.8"
thiserror = "1"
rayon = { version = "1.7", optional = true }
smartcore = { version = "0.3", optional = true }

[dev-dependencies]
approx = "0.5"
criterion = "0.5"

[features]
default = []
# Gates parallel fold execution (rayon) in cross_validate / learning_curve /
# validation_curve / nested_cv.
parallel = ["dep:rayon"]
# Gates the scoring::smartcore_adapter module wrapping smartcore::metrics.
smartcore-metrics = ["dep:smartcore"]

[[bench]]
name = "cross_validate"
harness = false

[package.metadata.docs.rs]
all-features = true