Created by: jerrywgz

Fix some typos in docs and get a better visualization. Add draw_threshold arg for inference.