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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
//! # RillML
//!
//! Lightweight, serializable online machine learning for Rust applications
//! and streaming data.
//!
//! RillML provides incremental learning primitives that can be embedded
//! directly in native Rust applications: online statistics, preprocessors,
//! linear/logistic regression, evaluation metrics, pipelines, progressive
//! evaluation, drift detection, online decision-making (bandits), and optional
//! serde-based state persistence.
//!
//! ## Quick start
//!
//! ```rust
//! use rill_ml::{
//! metrics::Mae,
//! models::{LinearRegression, LinearRegressionConfig},
//! optim::{Optimizer, SgdConfig},
//! pipeline::RegressionPipeline,
//! preprocessing::StandardScaler,
//! Metric, OnlineRegressor,
//! };
//!
//! let feature_count = 2;
//! let scaler = StandardScaler::new(feature_count).unwrap();
//! let optimizer = Optimizer::sgd(
//! feature_count,
//! SgdConfig { learning_rate: 0.05, l2: 0.0 },
//! ).unwrap();
//! let regression = LinearRegression::new(
//! feature_count,
//! LinearRegressionConfig { optimizer, loss: Default::default() },
//! ).unwrap();
//! let mut model = RegressionPipeline::new(scaler, regression).unwrap();
//! let mut mae = Mae::default();
//!
//! let samples = [
//! ([0.1, 0.2], 0.5),
//! ([0.3, 0.8], 1.4),
//! ([0.6, 0.4], 1.1),
//! ];
//! for (features, target) in samples {
//! let prediction = model.predict(&features).unwrap();
//! mae.update(target, prediction).unwrap();
//! model.learn(&features, target).unwrap();
//! }
//! ```
pub use RillError;
pub use ;
pub use ;