lgbm 0.0.6

Unofficial Rust bindings for LightGBM
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

use anyhow::Result;
use lgbm::{Dataset, Field, MatBuf, Parameters, parameters::Verbosity};
use std::env;

#[test]
fn from_csv() -> Result<()> {
    let csv_path = env::current_dir()?.join("tests/data/data.csv");
    let mut p = parameters();
    p.push("header", true);
    let d = Dataset::from_file(&csv_path, None, &p)?;
    assert_eq!(d.get_num_feature()?, 3);
    assert_eq!(d.get_num_data()?, 2);
    Ok(())
}

#[test]
fn from_mat_f32() -> Result<()> {
    let d = Dataset::from_mat(
        MatBuf::from_rows([[1.0f32, 2.0, 3.0], [4.0, 5.0, 6.0]]),
        None,
        &parameters(),
    )?;

    assert_eq!(d.get_num_feature()?, 3);
    assert_eq!(d.get_num_data()?, 2);
    Ok(())
}

#[test]
fn from_mat_f64() -> Result<()> {
    let d = Dataset::from_mat(
        MatBuf::from_rows([[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]]),
        None,
        &parameters(),
    )?;

    assert_eq!(d.get_num_feature()?, 3);
    assert_eq!(d.get_num_data()?, 2);
    Ok(())
}

#[test]
fn from_mats() -> Result<()> {
    let d = Dataset::from_mats(
        [
            MatBuf::from_rows([[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]]),
            MatBuf::from_rows([[1.0f64, 2.0, 3.0], [1.0f64, 2.0, 3.0], [1.0f64, 2.0, 3.0]]),
        ],
        None,
        &parameters(),
    )?;
    assert_eq!(d.get_num_feature()?, 3);
    assert_eq!(d.get_num_data()?, 5);
    Ok(())
}

#[test]
fn set_label() -> Result<()> {
    let mut d = Dataset::from_mat(
        MatBuf::from_rows([[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]]),
        None,
        &parameters(),
    )?;

    d.set_field(Field::LABEL, &[0.0, 1.0])?;
    Ok(())
}

#[test]
fn get_feature_names() -> Result<()> {
    let mut d = Dataset::from_mat(
        MatBuf::from_rows([[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]]),
        None,
        &parameters(),
    )?;

    let names0 = vec!["aaa", "bb", "c"];
    d.set_feature_names(&names0)?;
    let names1 = d.get_feature_names()?;
    assert_eq!(names0, names1);
    Ok(())
}

#[test]
fn set_feature_names_string() -> Result<()> {
    let mut d = Dataset::from_mat(
        MatBuf::from_rows([[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]]),
        None,
        &parameters(),
    )?;

    let names0 = vec!["aaa".to_string(), "bb".to_string(), "c".to_string()];
    d.set_feature_names(&names0)?;
    let names1 = d.get_feature_names()?;
    assert_eq!(names0, names1);
    Ok(())
}

fn parameters() -> Parameters {
    let mut p = Parameters::new();
    p.push("verbosity", Verbosity::Fatal);
    p
}