Machine learning in data modeling