updated machine learning documentation in README