Convolutional neutral network in Python