You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
673 B
Rust
20 lines
673 B
Rust
extern crate rusty_machine;
|
|
use rusty_machine::linalg::{Matrix,Vector};
|
|
use rusty_machine::learning::gp::{GaussianProcess,ConstMean};
|
|
use rusty_machine::learning::toolkit::kernel;
|
|
use rusty_machine::learning::SupModel;
|
|
|
|
fn main() {
|
|
let inputs = Matrix::new(3,3,vec![1.1,1.2,1.3,2.1,2.2,2.3,3.1,3.2,3.3]);
|
|
let targets = Vector::new(vec![0.1,0.8,0.3]);
|
|
|
|
let test_inputs = Matrix::new(2,3, vec![1.2,1.3,1.4,2.2,2.3,2.4]);
|
|
|
|
let ker = kernel::SquaredExp::new(2., 1.);
|
|
let zero_mean = ConstMean::default();
|
|
let mut gp = GaussianProcess::new(ker, zero_mean, 0.5);
|
|
|
|
gp.train(&inputs, &targets).unwrap();
|
|
let _ = gp.predict(&test_inputs).unwrap();
|
|
}
|