A new, blazing-fast learning engine for Continuous Time Bayesian Networks. Written in pure Rust. 🦀
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
use std::collections::BTreeSet;
|
|
|
|
|
|
|
|
use reCTBN::params;
|
|
|
|
|
|
|
|
#[allow(dead_code)]
|
|
|
|
pub fn generate_discrete_time_continous_node(label: String, cardinality: usize) -> params::Params {
|
|
|
|
params::Params::DiscreteStatesContinousTime(generate_discrete_time_continous_params(
|
|
|
|
label,
|
|
|
|
cardinality,
|
|
|
|
))
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn generate_discrete_time_continous_params(
|
|
|
|
label: String,
|
|
|
|
cardinality: usize,
|
|
|
|
) -> params::DiscreteStatesContinousTimeParams {
|
|
|
|
let domain: BTreeSet<String> = (0..cardinality).map(|x| x.to_string()).collect();
|
|
|
|
params::DiscreteStatesContinousTimeParams::new(label, domain)
|
|
|
|
}
|