Neural network for digit recognition
To install the package, run the command
pip install digit_recognition in the command prompt (windows) or terminal (mac) (alternative: download the package and run the setup.py script with
python setup.py install).
The package uses the following dependencies: numpy, urllib3, gzip, tkinter, Pillow. These should be installed automatically during the installation if necessary. If the package does not run as intended, please ensure that the dependencies listed here have been installed correctly.
1.2 Setting up the NN
The package does not include a trained NN, but only the necessary functions to build it. Therefore, prior to using the interface, the NN has to be set up (this has only to be done once). To do so, run the command
install_network in the command prompt/terminal (alternative: import the ‘digit_recognition’ module in python and call the
This creates a folder in the current directory (“DR_Data”) and downloads training and test sets from the MNIST database to this folder. It uses this data to train a NN with 784, 200, 100, and 10 nodes in each of four layers, respectively. The training algorithm goes through three epochs each with 60.000 training digits (this may take a few minutes). The NN is evaluated and the accuracy as well as the recall and precision for each digit are printed to the console (important: accuracy should be above 95%). Finally, the NN is saved to the “DR_Data” folder.
1.3 Starting up the interface
To start the interface, run the command
digit_recognition in the command prompt/terminal (alternative: import the ‘digit_recognition’ module in python and call the
Important: Ensure that the “DR_Data” folder is located in the current working directory.
1.4 Using the interface
The interface consists of two fields: a drawing field framed in black (left) and a feedback field showing several outputs (right). The user can draw in the drawing field by pressing the left mouse button and moving the mouse. Located below the drawing field are two buttons: the “Recognize!” button passes the drawing to the NN and displays its output in the feedback field; The “Reset” button deletes the current drawing and output. In the feedback field, three outputs are displayed: first, the digit recognized by the NN in the user’s drawing; second, the confidence of this recognition (i.e. the probability that the recognition is correct); and third, a possible alternative (i.e. the second most likely recognition). If the confidence is above 80%, no alternative is displayed.
Important: The performance of the NN is highly sensitive to size and location of the user’s digit in the drawing field. The grey rectangle in the drawing field indicates location and size for optimal performance.
2.1 NeuralNetwork (class)
__init__(self, design, weights=None, step_size=0.01, activation_function=sigmoid, bias=False) Set up basic attributes of neural network.
Attributes ---------- design: list Contains the number of nodes in each layer (length is number of layers) weights: list Contains weight matrices step_size: float Step size for training algorithm activation_function: function Activation function used in neural network bias: boolean Bias nodes on or off activation: list Contains activation of nodes at each layer confusion_matrix: np.array Confusion matrix produced in 'evaluate' method (true labels in rows, predicitons in columns) accuracy, recall, precision: float Accuracy, recall, and precision of neural network produced in 'evaluate' method Methods ------- train(input_data, target_data, epochs=1) Loops of 'one_training' function one_training(input_data, target_data) Computes cost and updates weights accordingly using backpropagation run(input_data) Forward propagation for single input evaluate(input_data, target_data) Assesses performance of neural network and computes performance measures save(file_name) Saves weights of neural network as np.array
train(self, input_data, target_data, epochs=1) Loop for 'one_train' (backpropagation) function (see below).
Arguments --------- input_data: array 3-dimensional numpy array containing input data (see pre_processing function for exact format). target_data: array 2-dimensional numpy array containing target data in one-hot representation. epochs: int Number of times the training algorithms iterates through the *entire* dataset. Defaults to 1.
one_training(self, input_data, target_data) Backpropagation algorithm to train neural network.
Arguments --------- input_data: array 2-dimensional numpy array containing a single input. Passed by train function (see above). target_data: array 2-dimensional numpy array containing a single target. Passed by train function (see above).
run(self, input_data) Forward propagation algorithm.
Computes output of neural network for a single input. Arguments --------- input_data: array 2-dimensional numpy array containing a single input.
evaluate(self, input_data, target_data) Evaluates performance of neural network. Computes accuracy of neural network, as well as recall and precision for each class using a confusion matrix. Results are printed to the console, but also defined as attributes of the neural network. Note: Use independent test data!
Arguments --------- input_data: array 3-dimensional numpy array containing test input data. target_data: array 2-dimensional numpy array containing test target data in one-hot representation.
save(self, file_name) Saves weights of neural network as np.array
Arguments --------- file_name: string Name of the file that is saved (without file extension)
2.2 Pre-processing (function)
pre_processing( ) Downloads, imports and preprocess data for digit recognition. Data is downloaded from the MNIST database. Consists of 70.000 handwritten digits of 28x28 pixels. Each with a corresponding, manually added label. Data is split into 60.000 instances for training and 10.000 instances for testing.
Returns: Matrix representations of digits and correspondings labels in a format optimized for the neural network.
# Import data train_images, train_labels, test_images, test_labels = pre_processing() # Build neural network (with two hidden layers of 200/100 nodes respectively) neural_network = NeuralNetwork([784,200,100,10], bias=True) neural_network.train(train_images, train_labels, epochs=3) neural_network.evaluate(test_images, test_labels) # Export neural network os.chdir('DR_Data') neural_network.save('my_network') os.chdir('..')
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size digit_recognition-0.0.11-py3-none-any.whl (14.1 kB)||File type Wheel||Python version py3||Upload date||Hashes View|
|Filename, size digit_recognition-0.0.11.tar.gz (9.3 kB)||File type Source||Python version None||Upload date||Hashes View|
Hashes for digit_recognition-0.0.11-py3-none-any.whl