rustyml 0.14.0

A high-performance machine learning & deep learning library in pure Rust, offering ML algorithms and neural network support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

//! Linear models for regression and classification
//!
//! Groups [`LinearRegression`] for regression and [`LogisticRegression`] for binary
//! classification. [`LinearRegression`] supports gradient-descent and closed-form solvers.
//! [`LogisticRegression`] trains only by gradient descent and adds the
//! [`generate_polynomial_features`] helper

/// Linear regression with gradient-descent and closed-form solvers
pub mod linear_regression;
/// Logistic regression for binary classification
pub mod logistic_regression;

pub use linear_regression::{LeastSquaresSolver, LinearRegression};
pub use logistic_regression::{LogisticRegression, generate_polynomial_features};