summaryrefslogtreecommitdiff
path: root/dtrain
diff options
context:
space:
mode:
Diffstat (limited to 'dtrain')
-rw-r--r--dtrain/dtrain.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/dtrain/dtrain.cc b/dtrain/dtrain.cc
index 79047fd9..ca5f0c5e 100644
--- a/dtrain/dtrain.cc
+++ b/dtrain/dtrain.cc
@@ -374,7 +374,7 @@ main(int argc, char** argv)
lambdas.plus_eq_v_times_s(diff_vec, eta);
rank_errors++;
}
- if (margin < 1) margin_violations++;
+ if (it->first.model - it->second.model < 1) margin_violations++;
} else {
// SVM
score_t margin = it->first.model - it->second.model;