Uncovering Atomic Shapes Through Machine Learning