@@ -155,7 +155,8 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
155
155
156
156
if classes. len ( ) != 2 {
157
157
return Err ( Failed :: fit ( & format ! (
158
- "Incorrect number of classes {}" , classes. len( )
158
+ "Incorrect number of classes {}" ,
159
+ classes. len( )
159
160
) ) ) ;
160
161
}
161
162
@@ -166,7 +167,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
166
167
if y_v != -T :: one ( ) || y_v != T :: one ( ) {
167
168
match y_v == classes[ 0 ] {
168
169
true => y. set ( i, -T :: one ( ) ) ,
169
- false => y. set ( i, T :: one ( ) )
170
+ false => y. set ( i, T :: one ( ) ) ,
170
171
}
171
172
}
172
173
}
@@ -194,7 +195,7 @@ impl<T: RealNumber, M: Matrix<T>, K: Kernel<T, M::RowVector>> SVC<T, M, K> {
194
195
for i in 0 ..n {
195
196
let cls_idx = match self . predict_for_row ( x. get_row ( i) ) == T :: one ( ) {
196
197
false => self . classes [ 0 ] ,
197
- true => self . classes [ 1 ]
198
+ true => self . classes [ 1 ] ,
198
199
} ;
199
200
y_hat. set ( i, cls_idx) ;
200
201
}
@@ -720,7 +721,8 @@ mod tests {
720
721
] ) ;
721
722
722
723
let y: Vec < f64 > = vec ! [
723
- -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ,
724
+ -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , -1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ,
725
+ 1. ,
724
726
] ;
725
727
726
728
let y_hat = SVC :: fit (
@@ -734,7 +736,7 @@ mod tests {
734
736
} ,
735
737
)
736
738
. and_then ( |lr| lr. predict ( & x) )
737
- . unwrap ( ) ;
739
+ . unwrap ( ) ;
738
740
739
741
assert ! ( accuracy( & y_hat, & y) >= 0.9 ) ;
740
742
}
0 commit comments