Machine learning probability model question