use-ml-model 0.0.1

ML model artifact metadata primitives for RustUse.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

use use_ml_model::{MlModelError, MlModelName, MlModelStage, MlModelTask};

#[test]
fn validates_model_metadata() -> Result<(), MlModelError> {
    let name = MlModelName::new("baseline-classifier")?;
    let task: MlModelTask = "classification".parse()?;

    assert_eq!(name.as_str(), "baseline-classifier");
    assert_eq!(task, MlModelTask::Classification);
    assert_eq!(MlModelStage::Experimental.as_str(), "experimental");
    Ok(())
}