Erratum to: Machine learning identification of symmetrized base states of Rydberg atoms