From 5890a353e6a158ff1b30a12565a6864f4c047f55 Mon Sep 17 00:00:00 2001
From: Rob Hess <hess@yahoo-inc.com>
Date: Fri, 13 Jun 2014 18:16:04 -0700
Subject: [PATCH] Add parameter for AccuracyLayer in proto.

---
 src/caffe/proto/caffe.proto | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto
index fd71f45..48d6fe2 100644
--- a/src/caffe/proto/caffe.proto
+++ b/src/caffe/proto/caffe.proto
@@ -188,6 +188,7 @@ message LayerParameter {
   // The weight decay that is multiplied on the global weight decay.
   repeated float weight_decay = 8;
 
+  optional AccuracyParameter accuracy_param = 27;
   optional ArgMaxParameter argmax_param = 23;
   optional ConcatParameter concat_param = 9;
   optional ConvolutionParameter convolution_param = 10;
@@ -214,6 +215,14 @@ message LayerParameter {
   optional V0LayerParameter layer = 1;
 }
 
+// Message that stores parameters used by AccuracyLayer
+message AccuracyParameter {
+  // When computing accuracy, count as correct by comparing the true label to
+  // the top k scoring classes.  By default, only compare to the top scoring
+  // class (i.e. argmax).
+  optional uint32 compare_to_top_k = 1 [default = 1];
+}
+
 // Message that stores parameters used by ArgMaxLayer
 message ArgMaxParameter {
   // If true produce pairs (argmax, maxval)
-- 
2.7.4