r2rs-stats 0.1.1

Statistics programming for Rust based on R's stats package
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
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

// "Whatever you do, work at it with all your heart, as working for the Lord,
// not for human masters, since you know that you will receive an inheritance
// from the Lord as a reward. It is the Lord Christ you are serving."
// (Col 3:23-24)

use polars::datatypes::DataType;
use strafe_datasets::baseball;
use strafe_type::ModelMatrix;

use super::tests::*;
use crate::regression::glm::family::gaussian::Gaussian;

#[test]
fn linear_regression_test() {
    linear_regression_test_inner(1, 1.0, 12.0, 8.0);
}

#[test]
fn generalized_linear_regression_test() {
    generalized_linear_regression_test_inner(1, 1.0, 12.0, 8.0);
}

#[test]
fn generalized_linear_regression_2() {
    let baseball = baseball().unwrap();

    let mut x_in = Vec::new();
    baseball
        .column("height")
        .unwrap()
        .cast(&DataType::Float64)
        .unwrap()
        .f64()
        .unwrap()
        .for_each(|f| x_in.push(f.unwrap()));
    let mut x_mm = ModelMatrix::from(&x_in);
    x_mm.set_name("x", "height");

    let mut y_in = Vec::new();
    baseball
        .column("weight")
        .unwrap()
        .cast(&DataType::Float64)
        .unwrap()
        .f64()
        .unwrap()
        .for_each(|f| y_in.push(f.unwrap()));
    let mut y_mm = ModelMatrix::from(&y_in);
    y_mm.set_name("x", "weight");

    fit_models(
        x_mm.matrix(),
        y_mm.matrix(),
        ModelMatrix::from(vec![1.0; x_in.len()]).matrix(),
        Gaussian::default(),
        &family_option_string("Gaussian"),
    );
}