linfa-linear 0.8.1

A Machine Learning framework for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

use std::error::Error;

use linfa::traits::Fit;
use linfa_linear::LinearRegression;

fn main() -> Result<(), Box<dyn Error>> {
    // load Diabetes dataset
    let dataset = linfa_datasets::diabetes();

    let lin_reg = LinearRegression::new();
    let model = lin_reg.fit(&dataset)?;

    println!("intercept:  {}", model.intercept());
    println!("parameters: {}", model.params());

    Ok(())
}