Make API a bit prettier

This commit is contained in:
2022-04-27 09:06:29 +02:00
parent aa8580970a
commit a8cef3806a
8 changed files with 42 additions and 32 deletions

View File

@@ -1,12 +1,9 @@
extern crate blas_src;
use kickscore::{
kernel::{self, Kernel},
model::{binary, Binary},
};
use kickscore::{kernel, model::Binary};
fn main() {
let mut model = Binary::new(binary::Observation::Probit);
let mut model = Binary::probit();
// Spike's skill does not change over time.
let k_spike = kernel::Constant::new(0.5);
@@ -15,10 +12,7 @@ fn main() {
let k_tom = kernel::Exponential::new(1.0, 1.0);
// Jerry's skill has a constant offset and smooth dynamics.
let k_jerry: [Box<dyn Kernel>; 2] = [
Box::new(kernel::Constant::new(1.0)),
Box::new(kernel::Matern52::new(0.5, 1.0)),
];
let k_jerry = (kernel::Constant::new(1.0), kernel::Matern52::new(0.5, 1.0));
// Now we are ready to add the items in the model.
model.add_item("Spike", k_spike);