Also predict for hosts.
Fix visualize.