mirror of
https://github.com/Magnus167/rustframe.git
synced 2025-08-20 23:40:01 +00:00
Compare commits
1 Commits
74060bb5c1
...
ddc4a71f76
Author | SHA1 | Date | |
---|---|---|---|
![]() |
ddc4a71f76 |
@ -59,7 +59,9 @@ impl GaussianNB {
|
|||||||
let bits = label.to_bits();
|
let bits = label.to_bits();
|
||||||
groups.entry(bits).or_default().push(i);
|
groups.entry(bits).or_default().push(i);
|
||||||
}
|
}
|
||||||
assert!(!groups.is_empty(), "No class labels found in y"); //-- panicked earlier
|
if groups.is_empty() {
|
||||||
|
panic!("No class labels found in y");
|
||||||
|
}
|
||||||
|
|
||||||
// Extract and sort class labels
|
// Extract and sort class labels
|
||||||
self.classes = groups.keys().cloned().map(f64::from_bits).collect();
|
self.classes = groups.keys().cloned().map(f64::from_bits).collect();
|
||||||
@ -207,5 +209,6 @@ mod tests {
|
|||||||
let y = Matrix::from_vec(vec![0.0], 1, 1);
|
let y = Matrix::from_vec(vec![0.0], 1, 1);
|
||||||
let mut clf = GaussianNB::new(1e-9, false);
|
let mut clf = GaussianNB::new(1e-9, false);
|
||||||
clf.fit(&x, &y);
|
clf.fit(&x, &y);
|
||||||
|
clf.predict(&x);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user