calculate the avg score for the model

[germs.git] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 90d3564d3d39fc08ae7fc48cb45610b146703f5d..b6194297d7e5f0d6290e64a09cca4dbf76d528a2 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -76,9 +76,12 @@ int main(int argc, char** argv){
        //TODO: Sort
        GeneSorter::ActionList al = so.sort(go,model);
 
+       double score = 0;
        for (GeneSorter::ActionList::iterator sa = al.begin(); sa != al.end(); ++sa){
-               cout << "Action: " << sa->toString() << " model score: " << model.score(*sa) << endl;
+               cout << "Action: " << sa->toString() << " model score: " << model.score(*sa,go) << endl;
+               score += model.score(*sa,go);
        }
+       cout << "Avg score: " << score / al.size() << endl;
        //TODO: Print result
        return EXIT_SUCCESS;
 }