1 files changed, 2 insertions, 0 deletions
diff --git a/train.cpp b/train.cpp
index 1c49902..fda06cb 100644
--- a/train.cpp
+++ b/train.cpp
@@ -84,6 +84,8 @@ int predict(DataProvider &dp) {
         vector<double> cur = cmc.getAcc();
         for (int i = 0;i<CMC_MAX;++i)
         *ot << cur[i]<<endl;
+        *ot << "AVG"<<endl;
+        *ot << cmc.getAvg()/D.getqSize() <<endl;
     }
     if (vm.count("output"))
         fout.close();