Skip to main content

ml_basic_usage/
basic_usage.rs

1use use_ml::{MlDatasetName, MlFeatureName, MlModelName, TensorShape};
2
3fn main() -> Result<(), Box<dyn std::error::Error>> {
4    let dataset = MlDatasetName::new("iris")?;
5    let feature = MlFeatureName::new("sepal_width")?;
6    let model = MlModelName::new("baseline-classifier")?;
7    let shape = TensorShape::new([150, 4])?;
8
9    assert_eq!(dataset.as_str(), "iris");
10    assert_eq!(feature.as_str(), "sepal_width");
11    assert_eq!(model.as_str(), "baseline-classifier");
12    assert_eq!(shape.dims(), &[150, 4]);
13    Ok(())
14}