diff options
| -rw-r--r-- | src/libMSVMMaj.c | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/libMSVMMaj.c b/src/libMSVMMaj.c index 65ff33d..ca0916b 100644 --- a/src/libMSVMMaj.c +++ b/src/libMSVMMaj.c @@ -70,9 +70,10 @@ void msvmmaj_category_matrix(struct MajModel *model, struct MajData *dataset) for (i=0; i<n; i++) { for (j=0; j<K; j++) { - if (dataset->y[i] != j+1) { + if (dataset->y[i] != j+1) matrix_set(model->R, K, i, j, 1.0); - } + else + matrix_set(model->R, K, i, j, 0.0); } } } |