def __init__(self): self.vocab = Index() for word in [ 'a', 'find', 'add', 'remove', 'clone', 'grass', 'red', 'wood', 'blue', 'clear', 'tree', 'house', 'wall', 'door', 'window', 'course', 'block', 'in', 'with', 'here', 'tall', 'short', ling.START, ling.STOP, ling.UNK ]: self.vocab.index(word)
def __init__(self): dir_path = os.path.dirname(os.path.realpath(__file__)) with open(os.path.join(dir_path, "recipes.yaml")) as recipes_f: recipes = yaml.load(recipes_f) # self.environment = set(recipes["environment"]) self.index = Index() self.environment = set(self.index.index(e) for e in recipes["environment"]) self.primitives = set(self.index.index(p) for p in recipes["primitives"]) self.recipes = {} for output, inputs in recipes["recipes"].items(): d = {} for inp, count in inputs.items(): # special keys if "_" in inp: d[inp] = count else: d[self.index.index(inp)] = count self.recipes[self.index.index(output)] = d kinds = self.environment | self.primitives | set(self.recipes.keys()) self.n_kinds = len(self.index)
def __init__(self, recipes_path): with open(recipes_path) as recipes_f: recipes = yaml.load(recipes_f) self.index = Index() self.environment = set( self.index.index(e) for e in recipes["environment"]) self.primitives = set( self.index.index(p) for p in recipes["primitives"]) self.recipes = {} for output, inputs in recipes["recipes"].items(): d = {} for inp, count in inputs.items(): # special keys if "_" in inp: d[inp] = count else: d[self.index.index(inp)] = count self.recipes[self.index.index(output)] = d self.kinds = self.environment | self.primitives | set( self.recipes.keys()) self.n_kinds = len(self.index)