fix indexing in cb_explore output

VowpalWabbit · Jun 17, 2016 · 038d17b · 038d17b
1 parent 8e0fd1c
commit 038d17b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/test/train-sets/ref/rcv1_raw_cb_explore.stderr b/test/train-sets/ref/rcv1_raw_cb_explore.stderr
@@ -10,16 +10,16 @@ Reading datafile = train-sets/rcv1_raw_cb_small.vw
 num sources = 1
 average  since         example        example  current  current  current
 loss     last          counter         weight    label  predict features
-0.050000 0.050000            1            1.0        1 0:0.975000      280
-0.025000 0.000000            2            2.0        1 1:0.975000      598
-0.012500 0.000000            4            4.0        1 1:0.975000      784
-0.250000 0.487500            8            8.0        1 1:0.975000      850
-0.368750 0.487500           16           16.0        1 1:0.975000      118
-0.245312 0.121875           32           32.0        1 0:0.975000      166
-0.216406 0.187500           64           64.0        2 0:0.975000      340
-0.216406 0.216406          128          128.0        1 1:0.975000      610
-0.300195 0.383984          256          256.0        2 1:0.975000      712
-0.337695 0.375195          512          512.0        2 0:0.975000      424
+0.050000 0.050000            1            1.0        1 1:0.975000      280
+0.025000 0.000000            2            2.0        1 2:0.975000      598
+0.012500 0.000000            4            4.0        1 2:0.975000      784
+0.250000 0.487500            8            8.0        1 2:0.975000      850
+0.368750 0.487500           16           16.0        1 2:0.975000      118
+0.245312 0.121875           32           32.0        1 1:0.975000      166
+0.216406 0.187500           64           64.0        2 1:0.975000      340
+0.216406 0.216406          128          128.0        1 2:0.975000      610
+0.300195 0.383984          256          256.0        2 2:0.975000      712
+0.337695 0.375195          512          512.0        2 1:0.975000      424
 
 finished run
 number of examples per pass = 1000

diff --git a/vowpalwabbit/cb_explore.cc b/vowpalwabbit/cb_explore.cc
@@ -312,7 +312,7 @@ namespace CB_EXPLORE{
       ss << temp_str;
       if(ec.pred.a_s[i].score > maxprob) {
 	maxprob = ec.pred.a_s[i].score;
-	maxid = i;
+	maxid = i+1;
       }
     }