Python Machine Learning Blueprints - Second Edition

Flattening

So far, we've focused on building as condensed and expressive a representation of our features as possible and used convolutional neural networks and max pooling layers to do this. The last step of our transformation is to flatten our convolved and max-pooled ndarray, in our example a 2 x 2 matrix, into a single row of training data.

Our max-pooled diagonal black line example would look something like the following, in code:

import numpy as npmax_pooled = np.array([[255,255],[255,255]])max_pooled

Running this code returns the following output:

We can check the shape here by running the following:

max_pooled.shape

This returns this ...