mirror of https://github.com/glouw/tinn
40 lines
1.1 KiB
C
40 lines
1.1 KiB
C
#pragma once
|
|
|
|
typedef struct
|
|
{
|
|
float* w; // All the weights.
|
|
float* x; // Hidden to output layer weights.
|
|
float* b; // Biases.
|
|
float* h; // Hidden layer.
|
|
float* o; // Output layer.
|
|
|
|
int nb; // Number of biases - always two - Tinn only supports a single hidden layer.
|
|
int nw; // Number of weights.
|
|
|
|
int nips; // Number of inputs.
|
|
int nhid; // Number of hidden neurons.
|
|
int nops; // Number of outputs.
|
|
}
|
|
Tinn;
|
|
|
|
// Trains a tinn with an input and target output with a learning rate.
|
|
// Returns error rate of the neural network.
|
|
float xttrain(const Tinn, const float* in, const float* tg, float rate);
|
|
|
|
// Builds a new tinn object given number of inputs (nips),
|
|
// number of hidden neurons for the hidden layer (nhid),
|
|
// and number of outputs (nops).
|
|
Tinn xtbuild(int nips, int nhid, int nops);
|
|
|
|
// Returns an output prediction given an input.
|
|
float* xpredict(const Tinn, const float* in);
|
|
|
|
// Saves the tinn to disk.
|
|
void xtsave(const Tinn, const char* path);
|
|
|
|
// Loads a new tinn from disk.
|
|
Tinn xtload(const char* path);
|
|
|
|
// Frees a tinn from the heap.
|
|
void xtfree(const Tinn);
|