ndarray-glm 0.1.0

Performs regression for generalized linear models using IRLS on data stored in arrays.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Handles edge cases that have caused trouble at times.
use anyhow::Result;
use ndarray::{Array2, array};
use ndarray_glm::{Logistic, ModelBuilder};
use num_traits::float::FloatCore;

/// Ensure that a valid likelihood is returned when the initial guess is the
/// best one.
#[test]
fn start_zero() -> Result<()> {
    // Exactly half of the data are true, meaning the initial guess of beta = 0 will be the best.
    let data_y = array![true, false, false, true];
    let data_x: Array2<f64> = array![[], [], [], []];
    let model = ModelBuilder::<Logistic>::data(&data_y, &data_x).build()?;
    let fit = model.fit()?;
    assert!(fit.model_like > -f64::infinity());

    Ok(())
}