spacecruft
/
karoo_gp
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Now with full Python 3.6 support!

tf2
Kai Staats 2019-06-08 22:28:38 -07:00 committed by GitHub
parent 877fc7ea38
commit c5ec0e9335
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 401 additions and 415 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

522
modules/karoo_gp_base_class.py
View File

File diff suppressed because it is too large Load Diff

294
modules/karoo_gp_pause.py
View File

@ -1,7 +1,7 @@
# Karoo GP Pause Menu # Karoo GP Pause Menu
# A text-based user interface for mid-run parameter configuration and population studies # A text-based user interface for mid-run parameter configuration and population studies
# by Kai Staats, MSc; see LICENSE.md # by Kai Staats, MSc with TensorFlow support provided by Iurii Milovanov; see LICENSE.md
# version 2.1.3 final for Python 2.7 # version 2.3 for Python 3.6
def pause(menu_dict): def pause(menu_dict):
@ -13,228 +13,224 @@ def pause(menu_dict):
Arguments required: menu_dict Arguments required: menu_dict
''' '''
options = ['','?','help','i','m','g','s','db','ts','min','bal','l','pop','e','p','id','dir','load','w','cont','q'] options = ['','?','help','i','m','g','s','db','ts','min','bal','l','pop','e','p','id','dir','load','w','add','q']
while True: while True:
try: try:
menu = raw_input('\n\t\033[36m (pause) \033[0;0m') menu = input('\n\t\033[36m (pause) \033[0;0m')
if menu not in options: raise ValueError() if menu in options: break
else: break else: raise ValueError()
except ValueError: print '\t\033[32m Select from the options given. Try again ...\033[0;0m' except ValueError: print ('\n\t\033[32m Enter \033[1m?\033[0;0m\033[32m to review your options. Try again ...\033[0;0m')
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
if menu == '': menu_dict['input_a'] = 'esc'; return menu_dict # bypass the following with ENTER if menu == '': menu_dict['input_a'] = 'esc' # exit (pause) with ENTER
elif menu == '?' or menu == 'help': elif menu == '?' or menu == 'help':
print '\n\t\033[36mSelect one of the following options:\033[0;0m' print ('\n\t\033[32m Select from one of the following options:\033[0;0m')
print '\t\033[36m\033[1m i \t\033[0;0m engage Interactive display mode' print ('\t\033[36m\033[1m i \t\033[0;0m engage Interactive display mode')
print '\t\033[36m\033[1m m \t\033[0;0m engage Minimal display mode' print ('\t\033[36m\033[1m m \t\033[0;0m engage Minimal display mode')
print '\t\033[36m\033[1m g \t\033[0;0m engage Generation display mode' print ('\t\033[36m\033[1m g \t\033[0;0m engage Generation display mode')
print '\t\033[36m\033[1m s \t\033[0;0m engage Silent display mode' print ('\t\033[36m\033[1m s \t\033[0;0m engage Silent display mode')
print '\t\033[36m\033[1m db \t\033[0;0m engage De-Bug display mode' print ('\t\033[36m\033[1m db \t\033[0;0m engage De-Bug display mode')
print '' print ('')
print '\t\033[36m\033[1m ts \t\033[0;0m adjust tournament size' print ('\t\033[36m\033[1m ts \t\033[0;0m adjust tournament size')
print '\t\033[36m\033[1m min \t\033[0;0m adjust minimum number of nodes' print ('\t\033[36m\033[1m min \t\033[0;0m adjust minimum number of nodes')
# print '\t\033[36m\033[1m max \t\033[0;0m adjust maximum Tree depth' # future option # print ('\t\033[36m\033[1m max \t\033[0;0m adjust maximum Tree depth') # NEED TO ADD
print '\t\033[36m\033[1m bal \t\033[0;0m adjust balance of genetic operators' print ('\t\033[36m\033[1m bal \t\033[0;0m adjust balance of genetic operators')
print '' print ('')
print '\t\033[36m\033[1m l \t\033[0;0m list Trees with leading fitness scores' print ('\t\033[36m\033[1m l \t\033[0;0m list Trees with leading fitness scores')
print '\t\033[36m\033[1m pop \t\033[0;0m list Trees in current population' print ('\t\033[36m\033[1m pop \t\033[0;0m list Trees in current population')
print '\t\033[36m\033[1m e \t\033[0;0m evaluate a single Tree against the test data' print ('\t\033[36m\033[1m e \t\033[0;0m evaluate a single Tree against the test data')
print '\t\033[36m\033[1m p \t\033[0;0m print a single Tree to screen' print ('\t\033[36m\033[1m p \t\033[0;0m print a single Tree to screen')
print '' print ('')
print '\t\033[36m\033[1m id \t\033[0;0m display current generation ID' print ('\t\033[36m\033[1m id \t\033[0;0m display current generation ID')
print '\t\033[36m\033[1m dir \t\033[0;0m display current working directory' print ('\t\033[36m\033[1m dir \t\033[0;0m display current working directory')
# print '\t\033[36m\033[1m load \t\033[0;0m load population_s (seed) to replace population_a (current)' # print ('\t\033[36m\033[1m load \t\033[0;0m load population_s (seed) to replace population_a (current)') # NEED TO FIX
print '\t\033[36m\033[1m w \t\033[0;0m write the evolving population_b to disk' print ('\t\033[36m\033[1m w \t\033[0;0m write the evolving population_b to disk')
print '' print ('')
print '\t\033[36m\033[1m cont \t\033[0;0m add generations and continue your run' print ('\t\033[36m\033[1m add \t\033[0;0m add generations and continue your run')
print '\t\033[36m\033[1m q \t\033[0;0m quit Karoo GP' print ('\t\033[36m\033[1m q \t\033[0;0m quit Karoo GP')
if menu_dict['gen_id'] == menu_dict['gen_max']: print '\n\t Your GP run is complete. You may \033[36m\033[1madd\033[0;0m generations and \033[36m\033[1mcont\033[0;0minue, or \033[36m\033[1mq\033[0;0muit.' elif menu == 'i': menu_dict['display'] = 'i'; print ('\n\t Interactive display mode engaged (for control freaks)')
else: print '\n\t You are mid-run. You may \033[36m\033[1m cont\033[0;0minue or \033[36m\033[1mq\033[0;0muit without saving population_b.' elif menu == 'g': menu_dict['display'] = 'g'; print ('\n\t Generation display mode engaged (for recovering control freaks)')
elif menu == 'm': menu_dict['display'] = 'm'; print ('\n\t Minimal display mode engaged (for GP gurus)')
elif menu == 'i': menu_dict['display'] = 'i'; print '\n\t Interactive display mode engaged (for control freaks)' elif menu == 's': menu_dict['display'] = 's'; print ('\n\t Silent display mode engaged (for zen masters)')
elif menu == 'm': menu_dict['display'] = 'm'; print '\n\t Minimal display mode engaged (for recovering control freaks)' elif menu == 'db': menu_dict['display'] = 'db'; print ('\n\t De-Bug display mode engaged (for evolutionary biologists)')
elif menu == 'g': menu_dict['display'] = 'g'; print '\n\t Generation display mode engaged (for GP gurus)'
elif menu == 's': menu_dict['display'] = 's'; print '\n\t Silent display mode engaged (for zen masters)'
elif menu == 'db': menu_dict['display'] = 'db'; print '\n\t De-Bug display mode engaged (for evolutionary biologists)'
elif menu == 'ts': # adjust the tournament size elif menu == 'ts': # adjust the tournament size
while True: while True:
try: try:
print '\n\t The current tournament size is:', menu_dict['tourn_size'] print ('\n\t The current tournament size is:', menu_dict['tourn_size'])
query = raw_input('\t Adjust the tournament size (suggest 7 for each 100): ') query = input('\t Adjust the tournament size (suggest 7 for each 100): ')
if query not in str(range(2,menu_dict['tree_pop_max'] + 1)) or query == '0' or query == '1': raise ValueError() # not ideal 20170918 if query == '': break
elif query == '': break elif int(query) in list(range(2,menu_dict['tree_pop_max'] + 1)): menu_dict['tourn_size'] = int(query); break # rebuilt 20190603
else: menu_dict['tourn_size'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 2 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']) except ValueError: print ('\n\t\033[32m Enter a number from 2 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
elif menu == 'min': # adjust the minimum number of nodes per Tree elif menu == 'min': # adjust the minimum number of nodes per Tree
# max_nodes = 2**(tree_depth_base +1) - 1 # NEED TO calc to replace upper limit in range but tree_depth_base is not global - 2018 04/22 # max_nodes = 2**(tree_depth_base +1) - 1 # NEED TO calc to replace upper limit in range but tree_depth_base is not global - 2018 04/22
while True: while True:
try: try:
print '\n\t The current minimum number of nodes is:', menu_dict['tree_depth_min'] print ('\n\t The current minimum number of nodes is:', menu_dict['tree_depth_min'])
query = raw_input('\t Adjust the minimum number of nodes for all Trees (min 3): ') query = input('\t Adjust the minimum number of nodes for all Trees (min 3): ')
if query not in str(range(3,1000)) or query == '0' or query == '1' or query == '2': raise ValueError() # not ideal 20170918 if query == '': break
elif query == '': break elif int(query) in list(range(3,1000)): menu_dict['tree_depth_min'] = int(query); break # rebuilt 20190603
else: menu_dict['tree_depth_min'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 3 including 1000. Try again ...\033[0;0m' except ValueError: print ('\n\t\033[32m Enter a number from 3 including 1000. Try again ...\033[0;0m')
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
# NEED TO ADD: adjustable tree_depth_max # NEED TO ADD
#elif menu == 'max': # adjust the maximum Tree depth #elif menu == 'max': # adjust the maximum Tree depth
#
# while True: # while True:
# try: # try:
# print '\n\t The current \033[3madjusted\033[0;0m maximum Tree depth is:', gp.tree_depth_max # print ('\n\t The current \033[3madjusted\033[0;0m maximum Tree depth is:', gp.tree_depth_max)
# query = raw_input('\n\t Adjust the global maximum Tree depth to (1 ... 10): ') # query = input('\n\t Adjust the global maximum Tree depth to (1 ... 10): ')
# if query not in str(range(1,11)): raise ValueError() # if int(query) not in list(range(1,11)): raise ValueError()
# if query < gp.tree_depth_max: # if query < gp.tree_depth_max:
# print '\n\t\033[32m This value is less than the current value.\033[0;0m' # print ('\n\t\033[32m This value is less than the current value.\033[0;0m')
# conf = raw_input('\n\t Are you ok with this? (y/n) ') # conf = input('\n\t Are you ok with this? (y/n) ')
# if conf == 'n': break # if conf == 'n': break
# except ValueError: print '\n\t\033[32m Enter a number from 1 including 10. Try again ...\033[0;0m' # except ValueError: print ('\n\t\033[32m Enter a number from 1 including 10. Try again ...\033[0;0m')
# except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' # except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
elif menu == 'bal': # adjust the balance of genetic operators' elif menu == 'bal': # adjust the balance of genetic operators'
print '\n\t The current balance of genetic operators is:' print ('\n\t The current balance of genetic operators is:')
print '\t\t Reproduction:', menu_dict['evolve_repro']; tmp_repro = menu_dict['evolve_repro'] print ('\t\t Reproduction:', menu_dict['evolve_repro']); tmp_repro = menu_dict['evolve_repro']
print '\t\t Point Mutation:', menu_dict['evolve_point']; tmp_point = menu_dict['evolve_point'] print ('\t\t Point Mutation:', menu_dict['evolve_point']); tmp_point = menu_dict['evolve_point']
print '\t\t Branch Mutation:', menu_dict['evolve_branch']; tmp_branch = menu_dict['evolve_branch'] print ('\t\t Branch Mutation:', menu_dict['evolve_branch']); tmp_branch = menu_dict['evolve_branch']
print '\t\t Crossover:', menu_dict['evolve_cross'], '\n'; tmp_cross = menu_dict['evolve_cross'] print ('\t\t Crossover:', menu_dict['evolve_cross'], '\n'); tmp_cross = menu_dict['evolve_cross']
while True: while True:
try: try:
query = raw_input('\t Enter quantity of Trees to be generated by Reproduction: ') query = input('\t Enter quantity of Trees to be generated by Reproduction: ')
if query not in str(range(0,1000)): raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(0,1000)): tmp_repro = int(query); break
else: tmp_repro = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']) except ValueError: print ('\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
while True: while True:
try: try:
query = raw_input('\t Enter quantity of Trees to be generated by Point Mutation: ') query = input('\t Enter quantity of Trees to be generated by Point Mutation: ')
if query not in str(range(0,1000)): raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(0,1000)): tmp_point = int(query); break
else: tmp_point = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']) except ValueError: print ('\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
while True: while True:
try: try:
query = raw_input('\t Enter quantity of Trees to be generated by Branch Mutation: ') query = input('\t Enter quantity of Trees to be generated by Branch Mutation: ')
if query not in str(range(0,1000)): raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(0,1000)): tmp_branch = int(query); break
else: tmp_branch = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']) except ValueError: print ('\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
while True: while True:
try: try:
query = raw_input('\t Enter quantity of Trees to be generated by Crossover: ') query = input('\t Enter quantity of Trees to be generated by Crossover: ')
if query not in str(range(0,1000)): raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(0,1000)): tmp_cross = int(query); break
else: tmp_cross = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']) except ValueError: print ('\n\t\033[32m Enter a number from 0 including %s. Try again ...\033[0;0m' %str(menu_dict['tree_pop_max']))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
if tmp_repro + tmp_point + tmp_branch + tmp_cross != menu_dict['tree_pop_max']: if tmp_repro + tmp_point + tmp_branch + tmp_cross != menu_dict['tree_pop_max']:
print '\n\t The sum of the above does not equal %s. Try again ...' %str(menu_dict['tree_pop_max']) print ('\n\t The sum of the above does not equal %s. Try again ...' %str(menu_dict['tree_pop_max']))
else: else:
print '\n\t The revised balance of genetic operators is:' print ('\n\t The revised balance of genetic operators is:')
print '\t\t Reproduction:', tmp_repro; menu_dict['evolve_repro'] = tmp_repro print ('\t\t Reproduction:', tmp_repro); menu_dict['evolve_repro'] = tmp_repro
print '\t\t Point Mutation:', tmp_point; menu_dict['evolve_point'] = tmp_point print ('\t\t Point Mutation:', tmp_point); menu_dict['evolve_point'] = tmp_point
print '\t\t Branch Mutation:', tmp_branch; menu_dict['evolve_branch'] = tmp_branch print ('\t\t Branch Mutation:', tmp_branch); menu_dict['evolve_branch'] = tmp_branch
print '\t\t Crossover:', tmp_cross; menu_dict['evolve_cross'] = tmp_cross print ('\t\t Crossover:', tmp_cross); menu_dict['evolve_cross'] = tmp_cross
elif menu == 'l': # display dictionary of Trees with the best fitness score elif menu == 'l': # display dictionary of Trees with the best fitness score
print '\n\t The leading Trees and their associated expressions are:' print ('\n\t The leading Trees and their associated expressions are:')
for n in sorted(menu_dict['fittest_dict']): print '\t ', n, ':', menu_dict['fittest_dict'][n] for n in sorted(menu_dict['fittest_dict']): print ('\t ', n, ':', menu_dict['fittest_dict'][n])
elif menu == 'pop': # list Trees in the current population elif menu == 'pop': # list Trees in the current population
if menu_dict['gen_id'] == 1: menu_dict['input_a'] = 'pop_a' if menu_dict['gen_id'] == 1: menu_dict['input_a'] = 'pop_a'
else: menu_dict['input_a'] = 'pop_b' else: menu_dict['input_a'] = 'pop_b'
elif menu == 'e': # evaluate a Tree against the TEST data elif menu == 'e': # evaluate a Tree against the TEST data
if menu_dict['gen_id'] == 1: print '\n\t\033[32m You cannot evaluate the foundation population. Be patient ...\033[0;0m' if menu_dict['gen_id'] == 1: print ('\n\t\033[32m You cannot evaluate the foundation population. Be patient ...\033[0;0m')
else: # gen_id > 1 else: # gen_id > 1
while True: while True:
try: try:
query = raw_input('\n\t Select a Tree to evaluate: ') query = input('\n\t Select a Tree to evaluate: ')
if query not in str(range(1, menu_dict['pop_b_len'])) or query == '0': raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(1, menu_dict['pop_b_len'])): menu_dict['input_a'] = 'eval'; menu_dict['input_b'] = int(query); break
else: menu_dict['input_a'] = 'eval'; menu_dict['input_b'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_b_len'] - 1) except ValueError: print ('\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_b_len'] - 1))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
elif menu == 'p': # print a Tree to screen -- NEED TO ADD: SymPy graphical print option elif menu == 'p': # print a Tree to screen -- NEED TO ADD: SymPy graphical print option
if menu_dict['gen_id'] == 1: if menu_dict['gen_id'] == 1: # first generation
while True: while True:
try: try:
query = raw_input('\n\t Select a Tree to print: ') query = input('\n\t Select a Tree to print: ')
if query not in str(range(1, menu_dict['pop_a_len'])) or query == '0': raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(1, menu_dict['pop_a_len'])) and menu_dict['gen_id'] == 1: menu_dict['input_a'] = 'print_a'; menu_dict['input_b'] = int(query); break
else: menu_dict['input_a'] = 'print_a'; menu_dict['input_b'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_a_len'] - 1) except ValueError: print ('\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_a_len'] - 1))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
else: # gen_id > 1 else: # second or higher generation
while True: while True:
try: try:
query = raw_input('\n\t Select a Tree to print: ') query = input('\n\t Select a Tree to print: ')
if query not in str(range(1, menu_dict['pop_b_len'])) or query == '0': raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(1, menu_dict['pop_b_len'])): menu_dict['input_a'] = 'print_b'; menu_dict['input_b'] = int(query); break
else: menu_dict['input_a'] = 'print_b'; menu_dict['input_b'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_b_len'] - 1) except ValueError: print ('\n\t\033[32m Enter a number from 1 including %s. Try again ...\033[0;0m' %str(menu_dict['pop_b_len'] - 1))
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
elif menu == 'id': print '\n\t The current generation is:', menu_dict['gen_id'] elif menu == 'id': print ('\n\t Current generation:', menu_dict['gen_id'])
elif menu == 'dir': print '\n\t The current working directory is:', menu_dict['path'] elif menu == 'dir': print ('\n\t Current working directory:', menu_dict['path'])
# NEED TO review and fix # NEED TO REBUILD
elif menu == 'load': # load population_s to replace population_a #elif menu == 'load': # load population_s to replace population_a
while True: # while True:
try: # try:
query = raw_input('\n\t Overwrite the current population with population_s? (y/n) ') # query = input('\n\t Overwrite the current population with population_s? (\033[1my\033[0;0m\033[32m/\033[1mn\033[0;0m\033[32m)\033[0;0m ')
if query not in ['y','n']: raise ValueError() # if query == 'y': menu_dict['input_a'] = 'load'; break
if query == 'y': menu_dict['input_a'] = 'load'; break # elif query == 'n': break
elif query == 'n': break # else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter (y)es or (n)o. Try again ...\033[0;0m' # except ValueError: print ('\n\t\033[32m Enter (\033[1my\033[0;0m)es or (\033[1mn\033[0;0m)o. Try again ...\033[0;0m')
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' # except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
elif menu == 'w': # write the evolving population_b to disk elif menu == 'w': # write the evolving population_b to disk
if menu_dict['gen_id'] > 1: menu_dict['input_a'] = 'write' if menu_dict['gen_id'] > 1: menu_dict['input_a'] = 'write'
else: print '\n\t\033[36m The evolving population_b does not yet exist\033[0;0m' else: print ('\n\t\033[36m The evolving population_b does not yet exist\033[0;0m')
elif menu == 'cont': # continue a GP run elif menu == 'add': # add generations and continue a GP run
if menu_dict['gen_id'] == menu_dict['gen_max']: if menu_dict['gen_id'] == menu_dict['gen_max']:
while True: while True:
try: try:
query = raw_input('\n\t\033[3m You are at the end of your run.\033[0;0m\n\t Add more generations to continue (1-100 or ENTER to escape): ') query = input('\n\t\033[3m You are at the end of your run.\033[0;0m\n\t Add more generations to continue (1-100 or ENTER to escape): ')
if query not in str(range(1,101)) or query == '0': raise ValueError() if query == '': break
elif query == '': break elif int(query) in list(range(1,101)): menu_dict['input_a'] = 'add'; menu_dict['input_b'] = int(query); break
else: menu_dict['input_a'] = 'cont'; menu_dict['input_b'] = int(query); break else: raise ValueError()
except ValueError: print '\n\t\033[32m Enter a number from 1 including 100. Try again ...\033[0;0m' except ValueError: print ('\n\t\033[32m Enter a number from 1 including 100. Try again ...\033[0;0m')
except KeyboardInterrupt: print '\n\t\033[32m Enter q to quit\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
else: menu_dict['input_a'] = 'cont' else: menu_dict['input_a'] = 'add'
elif menu == 'q': # quit elif menu == 'q': # quit (in case you didn't figure that one out :)
while True: while True:
try: try:
query = raw_input('\n\t Quit Karoo GP? (y/n) ') query = input('\n\t\033[32m Quit Karoo GP? (\033[1my\033[0;0m\033[32m/\033[1mn\033[0;0m\033[32m)\033[0;0m ')
if query not in ['y','n','']: raise ValueError()
if query == 'y': menu_dict['input_a'] = 'quit'; break if query == 'y': menu_dict['input_a'] = 'quit'; break
else: break else: break
except ValueError: print '\n\t\033[32m Enter (y)es or (n)o. Try again ...\033[0;0m' except ValueError: print ('\n\t\033[32m Enter \033[1my\033[0;0m\033[32mes or \033[1mn\033[0;0m\033[32mo\033[0;0m')
except KeyboardInterrupt: print '\n\t\033[32m Enter (y)es or (n)o. Try again ...\033[0;0m' except KeyboardInterrupt: print ('\n\n\t\033[32m Enter \033[1mq\033[0;0m\033[32m to quit\033[0;0m')
return menu_dict return menu_dict