(x, Q, title)
| 107 | return X[indexes], Y[indexes] |
| 108 | |
| 109 | def plot_image(x, Q, title): |
| 110 | im = Q.dot(x) |
| 111 | plt.imshow(im.reshape(28,28), cmap='gray') |
| 112 | plt.title(title) |
| 113 | plt.show() |
| 114 | |
| 115 | if __name__ == '__main__': |
| 116 | Xtrain = pd.read_csv('Xtrain.csv', header=None).as_matrix() |