datarust 0.6.0

Scikit-learn-style preprocessing and classical ML in Rust
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
//! Model-selection utilities mirroring `sklearn.model_selection`.
//!
//! Provides train/test splitting, K-fold cross-validation splitters, and a
//! generic [`cross_val_score`] that works with any [`Predictor`](crate::traits::Predictor)
//! + `Clone` estimator. A shared private PRNG lives in [`rng`].

/// Cross-validation scoring driver.
pub mod cross_val;
/// K-fold and stratified K-fold splitters.
pub mod kfold;
/// Shared deterministic PRNG (xorshift64).
pub mod rng;
/// Train/test split.
pub mod split;

pub use cross_val::cross_val_score;
pub use kfold::{KFold, StratifiedKFold};
pub use split::{train_test_split, TrainTestSplit};