Machine Learning Output Probability Distribution