Implement Display for NaiveBayes
This commit is contained in:
committed by
morenol
parent
ee6b6a53d6
commit
fabe362755
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user