week3_lm/homework.ipynb does averaging incorrectly and refers to rnn_lm instead of window_lm

[dniku]

It contains the following code:

assert np.mean(train_history[:10]) > np.mean(train_history[-10:]), "The model didn't converge."

This is incorrect because train_history is a list of pairs (i, loss). A correct way is:

assert np.mean(train_history[:10], axis=0)[1] > np.mean(train_history[-10:], axis=0)[1], "The model didn't converge."

Also, a couple of lines further it refers to rnn_lm instead of window_lm.

[justheuristic]

Thank you, wise scholar of the ways of spaghetti. We fixed it.