Learning political polarization on social media using neural networks