(x, Q, title)
| 90 | return X[indexes], Y[indexes] |
| 91 | |
| 92 | def plot_image(x, Q, title): |
| 93 | im = Q.dot(x) |
| 94 | plt.imshow(im.reshape(28,28), cmap='gray') |
| 95 | plt.title(title) |
| 96 | plt.show() |
| 97 | |
| 98 | if __name__ == '__main__': |
| 99 | Xtrain = pd.read_csv('Xtrain.csv', header=None).as_matrix() |