Understanding the desalination mechanism of a two-dimensional graphene-like membrane using data-driven design