Fmt.
This commit is contained in:
@@ -58,7 +58,7 @@ impl Variable {
|
||||
self.value = value;
|
||||
}
|
||||
|
||||
pub fn get_value(&self) -> Gaussian {
|
||||
pub fn get_value(&self) -> Gaussian {
|
||||
self.value
|
||||
}
|
||||
|
||||
@@ -101,7 +101,10 @@ impl PriorFactor {
|
||||
}
|
||||
|
||||
pub fn start(&self, variable_arena: &mut VariableArena) {
|
||||
variable_arena.get_mut(self.variable).unwrap().update_value(self.id, self.gaussian);
|
||||
variable_arena
|
||||
.get_mut(self.variable)
|
||||
.unwrap()
|
||||
.update_value(self.id, self.gaussian);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -146,7 +149,10 @@ impl LikelihoodFactor {
|
||||
|
||||
let gaussian = Gaussian::from_pi_tau(a * (x.pi() - fx.pi()), a * (x.tau() - fx.tau()));
|
||||
|
||||
variable_arena.get_mut(self.mean).unwrap().update_message(self.id, gaussian);
|
||||
variable_arena
|
||||
.get_mut(self.mean)
|
||||
.unwrap()
|
||||
.update_message(self.id, gaussian);
|
||||
}
|
||||
|
||||
pub fn update_value(&self, variable_arena: &mut VariableArena) {
|
||||
@@ -164,7 +170,10 @@ impl LikelihoodFactor {
|
||||
|
||||
let gaussian = Gaussian::from_pi_tau(a * (y.pi() - fy.pi()), a * (y.tau() - fy.tau()));
|
||||
|
||||
variable_arena.get_mut(self.value).unwrap().update_message(self.id, gaussian);
|
||||
variable_arena
|
||||
.get_mut(self.value)
|
||||
.unwrap()
|
||||
.update_message(self.id, gaussian);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -224,7 +233,10 @@ impl SumFactor {
|
||||
|
||||
let gaussian = Gaussian::from_pi_tau(new_pi, new_tau);
|
||||
|
||||
variable_arena.get_mut(variable).unwrap().update_message(self.id, gaussian);
|
||||
variable_arena
|
||||
.get_mut(variable)
|
||||
.unwrap()
|
||||
.update_message(self.id, gaussian);
|
||||
}
|
||||
|
||||
pub fn update_sum(&self, variable_arena: &mut VariableArena) {
|
||||
@@ -372,6 +384,9 @@ impl TruncateFactor {
|
||||
|
||||
let gaussian = Gaussian::from_pi_tau(c / m_w, (d + sqrt_c * v) / m_w);
|
||||
|
||||
variable_arena.get_mut(self.variable).unwrap().update_value(self.id, gaussian);
|
||||
variable_arena
|
||||
.get_mut(self.variable)
|
||||
.unwrap()
|
||||
.update_value(self.id, gaussian);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user