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 vote326 views1 answer
Colon in array python
Python
1 vote191 views2 answers
Convertir datetime to timestamp django
Python
42 votes217 views6 answers
Python comment
Python
21 votes205 views2 answers
Correct code to read csv file in python
Python
16 votes184 views2 answers
Python nonlocal
Python
0 vote193 views1 answer
Softmax function python
Python
5 votes179 views1 answer
Flask boiler plate
Python
0 vote187 views2 answers
Python verifier qu'un string est un chiffre
Python
0 vote189 views2 answers
Knapsack problem using greedy method in python
Python
60 votes176 views6 answers
Python remove duplicates from list
Python