First version.

1 files changed, 21 insertions, 1 deletions

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()