This line caused one of the problems with the multiclass demo, as the values were all 0. Switching to get_labels() fixed it, but the visualization is not that good (yet).