Commit aa6ebdf1 authored by Federico Rossi's avatar Federico Rossi

Added full time in report

parent d1cea0a4
......@@ -67,7 +67,7 @@ static void make_final(tiny_dnn::network<tiny_dnn::sequential> &nn,
}
void make_test_report(std::vector<tiny_dnn::vec_t>& imgs,int successes,std::vector<unsigned int> times,unsigned int total,std::string prefix) {
void make_test_report(std::vector<tiny_dnn::vec_t>& imgs,int successes,std::vector<unsigned int> times,unsigned int total,double meant,std::string prefix) {
std::stringstream ss;
ss << prefix;
#ifdef CNN_USE_POSIT
......@@ -85,6 +85,7 @@ void make_test_report(std::vector<tiny_dnn::vec_t>& imgs,int successes,std::vect
std::ofstream oft(filename+"_times.txt");
of << "Accuracy: " << (double)successes/(double)imgs.size() << std::endl;
of << "Time elapsed: " << total << std::endl;
of << "Mean t: " << meant << std::endl;
for(auto i: times) oft << i << std::endl;
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment