Plotting outputs for binary classifcation.
Adding multi classification implementation.