X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=src%2Fmodelidentifier.h;h=ee163e315d278d427ce646f8a52b29ec98052271;hb=43b1052d74a207fe667b75e41d0c70c1189c2cc8;hp=bd990443956fd1cb6809b8bd8792e3513d0ea942;hpb=90e2c43b99205b0d7a300209011231cd8d11f7a0;p=germs.git

diff --git a/src/modelidentifier.h b/src/modelidentifier.h
index bd99044..ee163e3 100644
--- a/src/modelidentifier.h
+++ b/src/modelidentifier.h
@@ -23,11 +23,13 @@
 
 #include "geneorder.h"
 
-#include <map>
+#include <queue>
 #include <string>
 #include <stdexcept>
 
 struct fann;
+class Model;
+
 
 /**
  * Identifies the model this gene order belongs to
@@ -35,7 +37,6 @@ struct fann;
  */
 class ModelIdentifier{
 	public:
-		enum Model{Whirl,X,FatX,Zipper,Cloud};
 
 		/**
 		 * Creates a new identifier given an artificial neural network
@@ -52,9 +53,7 @@ class ModelIdentifier{
 		 *
 		 * \returns a map with the model as key and the score between -1 and 1
 		 */
-		std::map<Model,double> identify(const GeneOrder& go);
-
-		static std::string modelName(Model m);
+		std::priority_queue<std::pair<double,Model> > identify(const GeneOrder& go);
 
 		~ModelIdentifier();
 	private: