Implement Display for NaiveBayes

This commit is contained in:
Lorenzo (Mec-iS)
2022-11-03 14:18:56 +00:00
committed by morenol
parent ee6b6a53d6
commit fabe362755
4 changed files with 62 additions and 0 deletions
+16
View File
@@ -271,6 +271,19 @@ pub struct GaussianNB<
inner: Option<BaseNaiveBayes<TX, TY, X, Y, GaussianNBDistribution<TY>>>,
}
impl<
TX: Number + RealNumber + RealNumber,
TY: Number + Ord + Unsigned,
X: Array2<TX>,
Y: Array1<TY>,
> fmt::Display for GaussianNB<TX, TY, X, Y>
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
writeln!(f, "GaussianNB:\ninner: {:?}", self.inner.as_ref().unwrap())?;
Ok(())
}
}
impl<
TX: Number + RealNumber + RealNumber,
TY: Number + Ord + Unsigned,
@@ -433,6 +446,9 @@ mod tests {
let gnb = GaussianNB::fit(&x, &y, parameters).unwrap();
assert_eq!(gnb.class_priors(), &priors);
// test display for GNB
println!("{}", &gnb);
}
#[cfg_attr(