Machine learning to remove space debris