def animate_fitting(tree, X, y, title, save_as=None):
tree.fit(X, y)
X = grid2d(100)
animation = AnimatedScatter(X, tree, title)
if save_as:
animation.ani.save(save_as, writer='imagemagick', fps=1)
animation.show()
dt = d[idx]["t"] - d[idx - 1]["t"]
t = (t - d[idx - 1]["t"]) / dt
pos = (posNext - posLast) * t + posLast
return pos
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("map", help="input file containing map")
parser.add_argument("schedule", help="schedule for agents")
parser.add_argument(
'--video',
dest='video',
default=None,
help="output video file (or leave empty to show on screen)")
parser.add_argument("--speed", type=int, default=1, help="speedup-factor")
args = parser.parse_args()
with open(args.map) as map_file:
map = yaml.load(map_file)
with open(args.schedule) as states_file:
schedule = yaml.load(states_file)
animation = Animation(map, schedule)
if args.video:
animation.save(args.video, args.speed)
else:
animation.show()