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()