Machine learning enabled endpoint security