Skip to content

Commit a9446c0

Browse files
Volodymyr OrlovVolodymyr Orlov
authored andcommitted
fix: fixes a bug in Eq implementation for SVC and SVR
1 parent 81395bc commit a9446c0

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/svm/svc.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
247247

248248
impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> PartialEq for SVC<T, M, K> {
249249
fn eq(&self, other: &Self) -> bool {
250-
if self.b != other.b
250+
if (self.b - other.b).abs() > T::epsilon() * T::two()
251251
|| self.w.len() != other.w.len()
252252
|| self.instances.len() != other.instances.len()
253253
{

src/svm/svr.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVR<T, M, K> {
204204

205205
impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> PartialEq for SVR<T, M, K> {
206206
fn eq(&self, other: &Self) -> bool {
207-
if self.b != other.b
207+
if (self.b - other.b).abs() > T::epsilon() * T::two()
208208
|| self.w.len() != other.w.len()
209209
|| self.instances.len() != other.instances.len()
210210
{

0 commit comments

Comments
 (0)