tinn/README.md
2018-04-10 18:53:29 -07:00

2.2 KiB

Tinn (Tiny Neural Network) is a 200 line dependency free neural network library written in C99.

#include "Tinn.h"
#include <stdio.h>

#define SETS 4
#define NIPS 2
#define NHID 8
#define NOPS 1
#define ITER 2000
#define RATE 1.0f

int main()
{
    // This example learns XOR.
    float in[SETS][NIPS] = {
        { 0, 0 },
        { 0, 1 },
        { 1, 0 },
        { 1, 1 },
    };
    float tg[SETS][NOPS] = {
        { 0 },
        { 1 },
        { 1 },
        { 0 },
    };
    // Build.
    const Tinn tinn = xtbuild(NIPS, NHID, NOPS);
    // Train.
    for(int i = 0; i < ITER; i++)
    {
        float error = 0.0f;
        for(int j = 0; j < SETS; j++)
            error += xttrain(tinn, in[j], tg[j], RATE);
        printf("%.12f\n", error / SETS);
    }
    // Predict.
    for(int i = 0; i < SETS; i++)
    {
        const float* pd = xtpredict(tinn, in[i]);
        printf("%f :: %f\n", tg[i][0], (double) pd[0]);
    }
    // Cleanup.
    xtfree(tinn);
    return 0;
}

For a more complicated demo on how to learn hand written digits, get some training data:

wget http://archive.ics.uci.edu/ml/machine-learning-databases/semeion/semeion.data

And if you're on Linux / MacOS just build and run Tinn with the test file:

make; ./tinn

If you're on Windows it's:

mingw32-make & tinn.exe

For the layman not accustomed to makefiles the makefile devolves into:

gcc test.c Tinn.c -lm

The training data consists of hand written digits written both slowly and quickly. Each line in the data set corresponds to one handwritten digit. Each digit is 16x16 pixels in size giving 256 inputs to the neural network.

At the end of the line 10 digits signify the digit:

0: 1 0 0 0 0 0 0 0 0 0
1: 0 1 0 0 0 0 0 0 0 0
2: 0 0 1 0 0 0 0 0 0 0
3: 0 0 0 1 0 0 0 0 0 0
4: 0 0 0 0 1 0 0 0 0 0
...
9: 0 0 0 0 0 0 0 0 0 1

This gives 10 outputs to the neural network. The test program will output the accuracy for each digit. Expect above 99% accuracy for the correct digit, and less that 1% accuracy for the other digits.

Disclaimer

Tinn is not a fully featured neural network C library like Kann, or Genann:

https://github.com/attractivechaos/kann

https://github.com/codeplea/genann

Tinn is minimal and easy to digest.