klave 0.5.0

A Rust SDK for the Klave platform.
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

use crate::sdk;

pub fn load_lightgbm_model(name: &str, model: &str) -> Result<(), Box<dyn std::error::Error>> {
    match sdk::load_lightgbm_model(name, model) {
        Ok(_) => Ok(()),
        Err(err) => Err(err.into()),
    }
}

pub fn unload_lightgbm_model(name: &str) -> Result<(), Box<dyn std::error::Error>> {
    match sdk::unload_lightgbm_model(name) {
        Ok(_) => Ok(()),
        Err(err) => Err(err.into()),
    }
}

pub fn infer_from_lightgbm_model(
    name: &str,
    data: Vec<f64>,
    nb_outputs: i32,
) -> Result<Vec<f64>, Box<dyn std::error::Error>> {
    match sdk::infer_from_lightgbm_model(name, &data, nb_outputs) {
        Ok(result) => Ok(result),
        Err(err) => Err(err.into()),
    }
}