From b08408d23186205e71dfc68634021e3236bfb45c Mon Sep 17 00:00:00 2001
From: InigoGutierrez <inigogf.95@gmail.com>
Date: Fri, 1 Jul 2022 16:10:15 +0200
Subject: First version.

---
 imagocli.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'imagocli.py')

diff --git a/imagocli.py b/imagocli.py
index e8e4580..5f14380 100755
--- a/imagocli.py
+++ b/imagocli.py
@@ -2,8 +2,28 @@
 
 """Run the Imago engine."""
 
+import sys
+
+from imago.data.enums import DecisionAlgorithms
 from imago.engine.imagoIO import ImagoIO
 
 if __name__ == "__main__":
-    io = ImagoIO()
+
+    decisionAlgorithm = None
+    if len(sys.argv) >= 3:
+        if sys.argv[1] == "-e":
+            if sys.argv[2] == "montecarlo":
+                decisionAlgorithm = DecisionAlgorithms.MONTECARLO
+            if sys.argv[2] == "keras":
+                decisionAlgorithm = DecisionAlgorithms.KERAS
+            if sys.argv[2] == "dense":
+                decisionAlgorithm = DecisionAlgorithms.DENSE
+            if sys.argv[2] == "conv":
+                decisionAlgorithm = DecisionAlgorithms.CONV
+
+    if decisionAlgorithm is None:
+        io = ImagoIO()
+    else:
+        io = ImagoIO(decisionAlgorithm)
+
     io.start()
-- 
cgit v1.2.1