Machine learning probability model answer