diff options
Diffstat (limited to 'imagocli.py')
-rwxr-xr-x | imagocli.py | 22 |
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() |