Information visualization using machine learning