Training Robots For Better Human-Robot Collaboration