Removed `node.rs` #45
Merged
AlessandroBregoli
merged 1 commits from 27-bug-useless-node
into dev
3 years ago
@ -1,25 +0,0 @@ |
||||
use crate::params::*; |
||||
|
||||
|
||||
pub struct Node { |
||||
pub params: Params, |
||||
pub label: String |
||||
} |
||||
|
||||
impl Node { |
||||
pub fn new(params: Params, label: String) -> Node { |
||||
Node{ |
||||
params: params, |
||||
label:label |
||||
} |
||||
} |
||||
|
||||
} |
||||
|
||||
impl PartialEq for Node { |
||||
fn eq(&self, other: &Node) -> bool{ |
||||
self.label == other.label |
||||
} |
||||
} |
||||
|
||||
|
@ -1,263 +1,365 @@ |
||||
mod utils; |
||||
use utils::*; |
||||
|
||||
use reCTBN::parameter_learning::*; |
||||
use ndarray::arr3; |
||||
use reCTBN::ctbn::*; |
||||
use reCTBN::network::Network; |
||||
use reCTBN::node; |
||||
use reCTBN::params; |
||||
use reCTBN::tools::*; |
||||
use ndarray::arr3; |
||||
use reCTBN::parameter_learning::*; |
||||
use reCTBN::{params, tools::*}; |
||||
use std::collections::BTreeSet; |
||||
|
||||
|
||||
#[macro_use] |
||||
extern crate approx; |
||||
|
||||
|
||||
fn learn_binary_cim<T: ParameterLearning> (pl: T) { |
||||
fn learn_binary_cim<T: ParameterLearning>(pl: T) { |
||||
let mut net = CtbnNetwork::new(); |
||||
let n1 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"),2)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"), 2)) |
||||
.unwrap(); |
||||
let n2 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"),2)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"), 2)) |
||||
.unwrap(); |
||||
net.add_edge(n1, n2); |
||||
|
||||
match &mut net.get_node_mut(n1).params { |
||||
match &mut net.get_node_mut(n1) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[[[-3.0, 3.0], [2.0, -2.0]]]))); |
||||
} |
||||
} |
||||
|
||||
match &mut net.get_node_mut(n2).params { |
||||
match &mut net.get_node_mut(n2) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[ |
||||
[[-1.0, 1.0], [4.0, -4.0]], |
||||
[[-6.0, 6.0], [2.0, -2.0]], |
||||
]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[ |
||||
[[-1.0, 1.0], [4.0, -4.0]], |
||||
[[-6.0, 6.0], [2.0, -2.0]], |
||||
])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
let data = trajectory_generator(&net, 100, 100.0, Some(6347747169756259),); |
||||
let data = trajectory_generator(&net, 100, 100.0, Some(6347747169756259)); |
||||
let (CIM, M, T) = pl.fit(&net, &data, 1, None); |
||||
print!("CIM: {:?}\nM: {:?}\nT: {:?}\n", CIM, M, T); |
||||
assert_eq!(CIM.shape(), [2, 2, 2]); |
||||
assert!(CIM.abs_diff_eq(&arr3(&[ |
||||
[[-1.0, 1.0], [4.0, -4.0]], |
||||
[[-6.0, 6.0], [2.0, -2.0]], |
||||
]), 0.1)); |
||||
assert!(CIM.abs_diff_eq( |
||||
&arr3(&[[[-1.0, 1.0], [4.0, -4.0]], [[-6.0, 6.0], [2.0, -2.0]],]), |
||||
0.1 |
||||
)); |
||||
} |
||||
|
||||
#[test] |
||||
fn learn_binary_cim_MLE() { |
||||
let mle = MLE{}; |
||||
let mle = MLE {}; |
||||
learn_binary_cim(mle); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_binary_cim_BA() { |
||||
let ba = BayesianApproach{ |
||||
alpha: 1, |
||||
tau: 1.0}; |
||||
let ba = BayesianApproach { alpha: 1, tau: 1.0 }; |
||||
learn_binary_cim(ba); |
||||
} |
||||
|
||||
fn learn_ternary_cim<T: ParameterLearning> (pl: T) { |
||||
fn learn_ternary_cim<T: ParameterLearning>(pl: T) { |
||||
let mut net = CtbnNetwork::new(); |
||||
let n1 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"), 3)) |
||||
.unwrap(); |
||||
let n2 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"), 3)) |
||||
.unwrap(); |
||||
net.add_edge(n1, n2); |
||||
|
||||
match &mut net.get_node_mut(n1).params { |
||||
match &mut net.get_node_mut(n1) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[[[-3.0, 2.0, 1.0],
|
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0]]]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[[ |
||||
[-3.0, 2.0, 1.0], |
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0] |
||||
]])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
match &mut net.get_node_mut(n2).params { |
||||
match &mut net.get_node_mut(n2) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
let data = trajectory_generator(&net, 100, 200.0, Some(6347747169756259),); |
||||
let data = trajectory_generator(&net, 100, 200.0, Some(6347747169756259)); |
||||
let (CIM, M, T) = pl.fit(&net, &data, 1, None); |
||||
print!("CIM: {:?}\nM: {:?}\nT: {:?}\n", CIM, M, T); |
||||
assert_eq!(CIM.shape(), [3, 3, 3]); |
||||
assert!(CIM.abs_diff_eq(&arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
]), 0.1)); |
||||
assert!(CIM.abs_diff_eq( |
||||
&arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
]), |
||||
0.1 |
||||
)); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_ternary_cim_MLE() { |
||||
let mle = MLE{}; |
||||
let mle = MLE {}; |
||||
learn_ternary_cim(mle); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_ternary_cim_BA() { |
||||
let ba = BayesianApproach{ |
||||
alpha: 1, |
||||
tau: 1.0}; |
||||
let ba = BayesianApproach { alpha: 1, tau: 1.0 }; |
||||
learn_ternary_cim(ba); |
||||
} |
||||
|
||||
fn learn_ternary_cim_no_parents<T: ParameterLearning> (pl: T) { |
||||
fn learn_ternary_cim_no_parents<T: ParameterLearning>(pl: T) { |
||||
let mut net = CtbnNetwork::new(); |
||||
let n1 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"), 3)) |
||||
.unwrap(); |
||||
let n2 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"), 3)) |
||||
.unwrap(); |
||||
net.add_edge(n1, n2); |
||||
|
||||
match &mut net.get_node_mut(n1).params { |
||||
match &mut net.get_node_mut(n1) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[[[-3.0, 2.0, 1.0],
|
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0]]]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[[ |
||||
[-3.0, 2.0, 1.0], |
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0] |
||||
]])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
match &mut net.get_node_mut(n2).params { |
||||
match &mut net.get_node_mut(n2) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
let data = trajectory_generator(&net, 100, 200.0, Some(6347747169756259),); |
||||
let data = trajectory_generator(&net, 100, 200.0, Some(6347747169756259)); |
||||
let (CIM, M, T) = pl.fit(&net, &data, 0, None); |
||||
print!("CIM: {:?}\nM: {:?}\nT: {:?}\n", CIM, M, T); |
||||
assert_eq!(CIM.shape(), [1, 3, 3]); |
||||
assert!(CIM.abs_diff_eq(&arr3(&[[[-3.0, 2.0, 1.0],
|
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0]]]), 0.1)); |
||||
assert!(CIM.abs_diff_eq( |
||||
&arr3(&[[[-3.0, 2.0, 1.0], [1.5, -2.0, 0.5], [0.4, 0.6, -1.0]]]), |
||||
0.1 |
||||
)); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_ternary_cim_no_parents_MLE() { |
||||
let mle = MLE{}; |
||||
let mle = MLE {}; |
||||
learn_ternary_cim_no_parents(mle); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_ternary_cim_no_parents_BA() { |
||||
let ba = BayesianApproach{ |
||||
alpha: 1, |
||||
tau: 1.0}; |
||||
let ba = BayesianApproach { alpha: 1, tau: 1.0 }; |
||||
learn_ternary_cim_no_parents(ba); |
||||
} |
||||
|
||||
|
||||
fn learn_mixed_discrete_cim<T: ParameterLearning> (pl: T) { |
||||
fn learn_mixed_discrete_cim<T: ParameterLearning>(pl: T) { |
||||
let mut net = CtbnNetwork::new(); |
||||
let n1 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n1"), 3)) |
||||
.unwrap(); |
||||
let n2 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"),3)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n2"), 3)) |
||||
.unwrap(); |
||||
|
||||
let n3 = net |
||||
.add_node(generate_discrete_time_continous_node(String::from("n3"),4)) |
||||
.add_node(generate_discrete_time_continous_node(String::from("n3"), 4)) |
||||
.unwrap(); |
||||
net.add_edge(n1, n2); |
||||
net.add_edge(n1, n3); |
||||
net.add_edge(n2, n3); |
||||
|
||||
match &mut net.get_node_mut(n1).params { |
||||
match &mut net.get_node_mut(n1) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[[[-3.0, 2.0, 1.0],
|
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0]]]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[[ |
||||
[-3.0, 2.0, 1.0], |
||||
[1.5, -2.0, 0.5], |
||||
[0.4, 0.6, -1.0] |
||||
]])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
match &mut net.get_node_mut(n2).params { |
||||
match &mut net.get_node_mut(n2) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.5], [3.0, -4.0, 1.0], [0.9, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 4.0], [1.5, -2.0, 0.5], [3.0, 1.0, -4.0]], |
||||
[[-1.0, 0.1, 0.9], [2.0, -2.5, 0.5], [0.9, 0.1, -1.0]], |
||||
])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
|
||||
match &mut net.get_node_mut(n3).params { |
||||
match &mut net.get_node_mut(n3) { |
||||
params::Params::DiscreteStatesContinousTime(param) => { |
||||
assert_eq!(Ok(()), param.set_cim(arr3(&[ |
||||
[[-1.0, 0.5, 0.3, 0.2], [0.5, -4.0, 2.5, 1.0], [2.5, 0.5, -4.0, 1.0], [0.7, 0.2, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 3.0, 1.0], [1.5, -3.0, 0.5, 1.0], [2.0, 1.3, -5.0 ,1.7], [2.5, 0.5, 1.0, -4.0]], |
||||
[[-1.3, 0.3, 0.1, 0.9], [1.4, -4.0, 0.5, 2.1], [1.0, 1.5, -3.0, 0.5], [0.4, 0.3, 0.1, -0.8]], |
||||
|
||||
[[-2.0, 1.0, 0.7, 0.3], [1.3, -5.9, 2.7, 1.9], [2.0, 1.5, -4.0, 0.5], [0.2, 0.7, 0.1, -1.0]], |
||||
[[-6.0, 1.0, 2.0, 3.0], [0.5, -3.0, 1.0, 1.5], [1.4, 2.1, -4.3, 0.8], [0.5, 1.0, 2.5, -4.0]], |
||||
[[-1.3, 0.9, 0.3, 0.1], [0.1, -1.3, 0.2, 1.0], [0.5, 1.0, -3.0, 1.5], [0.1, 0.4, 0.3, -0.8]], |
||||
|
||||
[[-2.0, 1.0, 0.6, 0.4], [2.6, -7.1, 1.4, 3.1], [5.0, 1.0, -8.0, 2.0], [1.4, 0.4, 0.2, -2.0]], |
||||
[[-3.0, 1.0, 1.5, 0.5], [3.0, -6.0, 1.0, 2.0], [0.3, 0.5, -1.9, 1.1], [5.0, 1.0, 2.0, -8.0]], |
||||
[[-2.6, 0.6, 0.2, 1.8], [2.0, -6.0, 3.0, 1.0], [0.1, 0.5, -1.3, 0.7], [0.8, 0.6, 0.2, -1.6]], |
||||
]))); |
||||
assert_eq!( |
||||
Ok(()), |
||||
param.set_cim(arr3(&[ |
||||
[ |
||||
[-1.0, 0.5, 0.3, 0.2], |
||||
[0.5, -4.0, 2.5, 1.0], |
||||
[2.5, 0.5, -4.0, 1.0], |
||||
[0.7, 0.2, 0.1, -1.0] |
||||
], |
||||
[ |
||||
[-6.0, 2.0, 3.0, 1.0], |
||||
[1.5, -3.0, 0.5, 1.0], |
||||
[2.0, 1.3, -5.0, 1.7], |
||||
[2.5, 0.5, 1.0, -4.0] |
||||
], |
||||
[ |
||||
[-1.3, 0.3, 0.1, 0.9], |
||||
[1.4, -4.0, 0.5, 2.1], |
||||
[1.0, 1.5, -3.0, 0.5], |
||||
[0.4, 0.3, 0.1, -0.8] |
||||
], |
||||
[ |
||||
[-2.0, 1.0, 0.7, 0.3], |
||||
[1.3, -5.9, 2.7, 1.9], |
||||
[2.0, 1.5, -4.0, 0.5], |
||||
[0.2, 0.7, 0.1, -1.0] |
||||
], |
||||
[ |
||||
[-6.0, 1.0, 2.0, 3.0], |
||||
[0.5, -3.0, 1.0, 1.5], |
||||
[1.4, 2.1, -4.3, 0.8], |
||||
[0.5, 1.0, 2.5, -4.0] |
||||
], |
||||
[ |
||||
[-1.3, 0.9, 0.3, 0.1], |
||||
[0.1, -1.3, 0.2, 1.0], |
||||
[0.5, 1.0, -3.0, 1.5], |
||||
[0.1, 0.4, 0.3, -0.8] |
||||
], |
||||
[ |
||||
[-2.0, 1.0, 0.6, 0.4], |
||||
[2.6, -7.1, 1.4, 3.1], |
||||
[5.0, 1.0, -8.0, 2.0], |
||||
[1.4, 0.4, 0.2, -2.0] |
||||
], |
||||
[ |
||||
[-3.0, 1.0, 1.5, 0.5], |
||||
[3.0, -6.0, 1.0, 2.0], |
||||
[0.3, 0.5, -1.9, 1.1], |
||||
[5.0, 1.0, 2.0, -8.0] |
||||
], |
||||
[ |
||||
[-2.6, 0.6, 0.2, 1.8], |
||||
[2.0, -6.0, 3.0, 1.0], |
||||
[0.1, 0.5, -1.3, 0.7], |
||||
[0.8, 0.6, 0.2, -1.6] |
||||
], |
||||
])) |
||||
); |
||||
} |
||||
} |
||||
|
||||
|
||||
let data = trajectory_generator(&net, 300, 300.0, Some(6347747169756259),); |
||||
let data = trajectory_generator(&net, 300, 300.0, Some(6347747169756259)); |
||||
let (CIM, M, T) = pl.fit(&net, &data, 2, None); |
||||
print!("CIM: {:?}\nM: {:?}\nT: {:?}\n", CIM, M, T); |
||||
assert_eq!(CIM.shape(), [9, 4, 4]); |
||||
assert!(CIM.abs_diff_eq(&arr3(&[ |
||||
[[-1.0, 0.5, 0.3, 0.2], [0.5, -4.0, 2.5, 1.0], [2.5, 0.5, -4.0, 1.0], [0.7, 0.2, 0.1, -1.0]], |
||||
[[-6.0, 2.0, 3.0, 1.0], [1.5, -3.0, 0.5, 1.0], [2.0, 1.3, -5.0 , 1.7], [2.5, 0.5, 1.0, -4.0]], |
||||
[[-1.3, 0.3, 0.1, 0.9], [1.4, -4.0, 0.5, 2.1], [1.0, 1.5, -3.0, 0.5], [0.4, 0.3, 0.1, -0.8]], |
||||
|
||||
[[-2.0, 1.0, 0.7, 0.3], [1.3, -5.9, 2.7, 1.9], [2.0, 1.5, -4.0, 0.5], [0.2, 0.7, 0.1, -1.0]], |
||||
[[-6.0, 1.0, 2.0, 3.0], [0.5, -3.0, 1.0, 1.5], [1.4, 2.1, -4.3, 0.8], [0.5, 1.0, 2.5, -4.0]], |
||||
[[-1.3, 0.9, 0.3, 0.1], [0.1, -1.3, 0.2, 1.0], [0.5, 1.0, -3.0, 1.5], [0.1, 0.4, 0.3, -0.8]], |
||||
|
||||
[[-2.0, 1.0, 0.6, 0.4], [2.6, -7.1, 1.4, 3.1], [5.0, 1.0, -8.0, 2.0], [1.4, 0.4, 0.2, -2.0]], |
||||
[[-3.0, 1.0, 1.5, 0.5], [3.0, -6.0, 1.0, 2.0], [0.3, 0.5, -1.9, 1.1], [5.0, 1.0, 2.0, -8.0]], |
||||
[[-2.6, 0.6, 0.2, 1.8], [2.0, -6.0, 3.0, 1.0], [0.1, 0.5, -1.3, 0.7], [0.8, 0.6, 0.2, -1.6]], |
||||
]), 0.1)); |
||||
assert!(CIM.abs_diff_eq( |
||||
&arr3(&[ |
||||
[ |
||||
[-1.0, 0.5, 0.3, 0.2], |
||||
[0.5, -4.0, 2.5, 1.0], |
||||
[2.5, 0.5, -4.0, 1.0], |
||||
[0.7, 0.2, 0.1, -1.0] |
||||
], |
||||
[ |
||||
[-6.0, 2.0, 3.0, 1.0], |
||||
[1.5, -3.0, 0.5, 1.0], |
||||
[2.0, 1.3, -5.0, 1.7], |
||||
[2.5, 0.5, 1.0, -4.0] |
||||
], |
||||
[ |
||||
[-1.3, 0.3, 0.1, 0.9], |
||||
[1.4, -4.0, 0.5, 2.1], |
||||
[1.0, 1.5, -3.0, 0.5], |
||||
[0.4, 0.3, 0.1, -0.8] |
||||
], |
||||
[ |
||||
[-2.0, 1.0, 0.7, 0.3], |
||||
[1.3, -5.9, 2.7, 1.9], |
||||
[2.0, 1.5, -4.0, 0.5], |
||||
[0.2, 0.7, 0.1, -1.0] |
||||
], |
||||
[ |
||||
[-6.0, 1.0, 2.0, 3.0], |
||||
[0.5, -3.0, 1.0, 1.5], |
||||
[1.4, 2.1, -4.3, 0.8], |
||||
[0.5, 1.0, 2.5, -4.0] |
||||
], |
||||
[ |
||||
[-1.3, 0.9, 0.3, 0.1], |
||||
[0.1, -1.3, 0.2, 1.0], |
||||
[0.5, 1.0, -3.0, 1.5], |
||||
[0.1, 0.4, 0.3, -0.8] |
||||
], |
||||
[ |
||||
[-2.0, 1.0, 0.6, 0.4], |
||||
[2.6, -7.1, 1.4, 3.1], |
||||
[5.0, 1.0, -8.0, 2.0], |
||||
[1.4, 0.4, 0.2, -2.0] |
||||
], |
||||
[ |
||||
[-3.0, 1.0, 1.5, 0.5], |
||||
[3.0, -6.0, 1.0, 2.0], |
||||
[0.3, 0.5, -1.9, 1.1], |
||||
[5.0, 1.0, 2.0, -8.0] |
||||
], |
||||
[ |
||||
[-2.6, 0.6, 0.2, 1.8], |
||||
[2.0, -6.0, 3.0, 1.0], |
||||
[0.1, 0.5, -1.3, 0.7], |
||||
[0.8, 0.6, 0.2, -1.6] |
||||
], |
||||
]), |
||||
0.1 |
||||
)); |
||||
} |
||||
|
||||
#[test] |
||||
fn learn_mixed_discrete_cim_MLE() { |
||||
let mle = MLE{}; |
||||
let mle = MLE {}; |
||||
learn_mixed_discrete_cim(mle); |
||||
} |
||||
|
||||
|
||||
#[test] |
||||
fn learn_mixed_discrete_cim_BA() { |
||||
let ba = BayesianApproach{ |
||||
alpha: 1, |
||||
tau: 1.0}; |
||||
let ba = BayesianApproach { alpha: 1, tau: 1.0 }; |
||||
learn_mixed_discrete_cim(ba); |
||||
} |
||||
|
@ -1,16 +1,17 @@ |
||||
use reCTBN::params; |
||||
use reCTBN::node; |
||||
use std::collections::BTreeSet; |
||||
|
||||
pub fn generate_discrete_time_continous_node(name: String, cardinality: usize) -> node::Node { |
||||
node::Node::new(params::Params::DiscreteStatesContinousTime(generate_discrete_time_continous_param(cardinality)), name) |
||||
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_param(cardinality: usize) -> params::DiscreteStatesContinousTimeParams{ |
||||
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(domain) |
||||
params::DiscreteStatesContinousTimeParams::new(label, domain) |
||||
} |
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
Loading…
Reference in new issue