Generate training set and Visualization