<s> import matplotlib . pyplot as plt <EOL> import theanets <EOL> from utils import load_mnist , plot_layers <EOL> train , valid , _ = load_mnist ( labels = True ) <EOL> N = <NUM_LIT:10> <EOL> net = theanets . Classifier ( [ <NUM_LIT> , N * N , ( '<STR_LIT>' , <NUM_LIT:10> ) ] ) <EOL> net . train ( train , valid , min_improvement = <NUM_LIT> , train_batches = <NUM_LIT:100> ) <EOL> plot_layers ( [ net . find ( '<STR_LIT>' , '<STR_LIT:w>' ) , net . find ( '<STR_LIT>' , '<STR_LIT:w>' ) ] ) <EOL> plt . tight_layout ( ) <EOL> plt . show ( ) </s>
