randomforest

randomforest Documentation Actions Status Coverage Status License: MIT

A random forest implementation in Rust.

Examples

```rust use randomforest::criterion::Mse; use randomforest::RandomForestRegressorOptions; use randomforest::table::TableBuilder;

let features = [ &[0.0, 2.0, 1.0, 0.0][..], &[0.0, 2.0, 1.0, 1.0][..], &[1.0, 2.0, 1.0, 0.0][..], &[2.0, 1.0, 1.0, 0.0][..], &[2.0, 0.0, 0.0, 0.0][..], &[2.0, 0.0, 0.0, 1.0][..], &[1.0, 0.0, 0.0, 1.0][..], &[0.0, 1.0, 1.0, 0.0][..], &[0.0, 0.0, 0.0, 0.0][..], &[2.0, 1.0, 0.0, 0.0][..], &[0.0, 1.0, 0.0, 1.0][..], &[1.0, 1.0, 1.0, 1.0][..],

]; let target = [ 25.0, 30.0, 46.0, 45.0, 52.0, 23.0, 43.0, 35.0, 38.0, 46.0, 48.0, 52.0 ];

let mut tablebuilder = TableBuilder::new(); for (xs, y) in features.iter().zip(target.iter()) { tablebuilder.addrow(xs, *y)?; } let table = tablebuilder.build()?;

let regressor = RandomForestRegressorOptions::new() .seed(0) .fit(Mse, table); assert_eq!(regressor.predict(&[1.0, 2.0, 0.0, 0.0]), 41.9785); ```