Machine learning in scala