Machine learning integration