saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up code (#1939)

Browse source
This commit is contained in:
Changming Sun 2019-09-27 10:17:46 -07:00 committed by GitHub
parent 1649374a5c
commit 174c142eff
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
onnxruntime/core/providers/cpu/ml/ml_common.h
View file

@ -322,13 +322,10 @@ void write_scores(std::vector<T>& scores, POST_EVAL_TRANSFORM post_transform, in
scores[0] = ComputeProbit(scores[0]);
} else {
switch (add_second_class) {
case 0: //0=all positive weights, winning class is positive
case 0:
case 1:
scores.push_back(scores[0]);
scores[0] = 1.f - scores[0]; //put opposite score in positive slot
break;
case 1: //1 = all positive weights, winning class is negative
scores.push_back(scores[0]);
scores[0] = 1.f - scores[0]; //put opposite score in positive slot
scores[0] = 1.f - scores[0];
break;
case 2: //2 = mixed weights, winning class is positive
if (post_transform == POST_EVAL_TRANSFORM::LOGISTIC) {