Mnist

import tflearn
from tflearn.layers.conv import conv_2d, max_pool_2d
from tflearn.layers.core import input_data, dropout, fully_connected
from tflearn.layers.estimator import regression
import tflearn.datasets.mnist as mnist

X, Y, test_x, test_y = mnist.load_data(one_hot=True)

X = X.reshape([-1, 28, 28, 1])
test_x = test_x.reshape([-1, 28, 28, 1])

convnet = input_data(shape=[None, 28, 28, 1], name='input')

convnet = conv_2d(convnet, 32, 2, activation='relu')
convnet = max_pool_2d(convnet, 2)

convnet = conv_2d(convnet, 64, 2, activation='relu')
convnet = max_pool_2d(convnet, 2)

convnet = fully_connected(convnet, 1024, activation='relu')
convnet = dropout(convnet, 0.8)

convnet = fully_connected(convnet, 10, activation='softmax')
convnet = regression(convnet, optimizer='adam', learning_rate=0.01, loss='categorical_crossentropy', name='targets')

model = tflearn.DNN(convnet)
model.fit({'input': X}, {'targets': Y}, n_epoch=10, validation_set=({'input': test_x}, {'targets': test_y}), 
    snapshot_step=500, show_metric=True, run_id='mnist')

Source: pythonprogramming.net

Similar codes

1 vote239 views1 answer
, in <genexpr> if not all (key in json for key in transaction_keys): TypeError: argument of type 'NoneType' is not iterable
Python
0 vote225 views1 answer
Pytest check exception
Python
2 votes225 views1 answer
Convert python pandas series dtype to datetime
Python
0 vote239 views1 answer
String times python
Python
0 vote231 views1 answer
Get data from model with field name in django
Python Django
0 vote236 views1 answer
Raspberry pi pygame play thru bluetooth device
Python
4 votes220 views4 answers
Jupyter lab autocomplete
Python
6 votes233 views3 answers
Messages django
Python Django
0 vote228 views1 answer
How to stop all pygame mixer sound
Python
0 vote235 views1 answer
Install aws sdk ubuntu 20.04 command line
Python