linfa-elasticnet 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
18
19
20
21
22
23
24

use linfa::prelude::*;
use linfa_elasticnet::{MultiTaskElasticNet, Result};

fn main() -> Result<()> {
    // load Diabetes dataset
    let (train, valid) = linfa_datasets::linnerud().split_with_ratio(0.80);

    // train pure LASSO model with 0.1 penalty
    let model = MultiTaskElasticNet::params()
        .penalty(0.1)
        .l1_ratio(1.0)
        .fit(&train)?;

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

    println!("z score: {:?}", model.z_score());

    // validate
    let y_est = model.predict(&valid);
    println!("predicted variance: {}", y_est.r2(&valid)?);

    Ok(())
}