Machine learning for productivity