def enumerateDreams(checkpoint, directory):
from dreamcoder.dreaming import backgroundHelmholtzEnumeration
from dreamcoder.utilities import loadPickle
result = loadPickle(checkpoint)
eprint(" [+] Loaded checkpoint",checkpoint)
g = result.grammars[-1]
if directory is None: assert False, "please specify a directory"
eprint(" Dreaming into",directory)
os.system("mkdir -p %s"%directory)
frontiers = backgroundHelmholtzEnumeration(makeTasks(None,None), g, 100,
evaluationTimeout=0.01,
special=LogoFeatureCNN.special)()
print(f"{len(frontiers)} total frontiers.")
MDL = 0
def L(f):
return -list(f.entries)[0].logPrior
frontiers.sort(key=lambda f: -L(f))
while len(frontiers) > 0:
# get frontiers whose MDL is between [MDL,MDL + 1)
fs = []
while len(frontiers) > 0 and L(frontiers[-1]) < MDL + 1:
fs.append(frontiers.pop(len(frontiers) - 1))
if fs:
random.shuffle(fs)
print(f"{len(fs)} programs with MDL between [{MDL}, {MDL + 1})")
fs = fs[:500]
os.system(f"mkdir {directory}/{MDL}")
dreamFromGrammar([list(f.entries)[0].program for f in fs],
f"{directory}/{MDL}")
MDL += 1
def outputDreams(checkpoint, directory):
from dreamcoder.utilities import loadPickle
result = loadPickle(checkpoint)
eprint(" [+] Loaded checkpoint", checkpoint)
g = result.grammars[-1]
if directory is None:
randomStr = ''.join(random.choice('0123456789') for _ in range(10))
directory = "/tmp/" + randomStr
eprint(" Dreaming into", directory)
os.system("mkdir -p %s" % directory)
dreamFromGrammar(g, directory)