/Users/oliviasalomon/miniconda3/envs/iaml/lib/python2.7/site-packages/ipykernel_launcher.py:21: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-90-7f03f12dfcc7> in <module>()
     19
     20 for color, i, target_name in zip(colors, [0, 1, 2, 3, 4], classes):
---> 21     plt.scatter(X_r[y == i, 0], X_r[y == i, 1], color=color, alpha=.8, lw=lw,
     22                 label=target_name)
     23 plt.legend(loc='best', shadow=False, scatterpoints=1)

IndexError: in the future, 0-d boolean arrays will be interpreted as a valid boolean index

<matplotlib.figure.Figure at 0x10b2ba5d0>
^I promise that THIS WAS WORKING in my previous submission, but it's no longer working! I believe that I accidentally deleted the line where I defined y2, but I can't remember how I made it work, and I forgot to continuously save and so cannot revert to an earlier version to see.

I had to resubmit the assignment because I've added a lot of work to 2.5, and I figured that 2.5 is worth more than this problem, so I'm making this sacrifice. I'm sorry it's not working anymore :(