Polarization-based underwater geolocalization with neural network models