1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#ifndef __NN__
#define __NN__
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <math.h>
#include <unistd.h>
#include <openblas/cblas.h>
typedef struct Layer {
double *weights, *bias;
double (*activation)(double x);
size_t neurons, input_nodes;
} Layer;
void nn_network_init_weights(Layer *network, size_t nmemb, size_t input_cols);
void nn_network_free_weights(Layer *network, size_t nmemb);
void nn_layer_map_activation(
double (*activation)(double),
double *aout, size_t aout_shape[2],
double *zout, size_t zout_shape[2]);
void nn_layer_forward(Layer layer, double *out, size_t out_shape[2], double *input, size_t input_shape[2]);
void nn_layer_backward(
Layer layer,
double *weights,
double *cost_derivative, size_t dcost_shape[2],
double *out, size_t out_shape[2],
double *labels, size_t labels_shape[2],
double *local_gradient); //TODO
double sigmoid(double x);
double relu(double x);
double identity(double x);
void nn_forward(double **aout, double **zout, double *input, size_t input_shape[2], Layer network[], size_t network_size);
void nn_layer_out_delta(
double *delta, size_t delta_cols,
double *error, size_t error_cols,
double *zout, size_t zout_cols,
double (*activation_derivative)(double));//TODO
void nn_layer_hidden_delta(
double *delta, size_t delta_cols,
double *delta_next, size_t delta_next_cols,
double *weigths_next, size_t weigths_shape[2],
double *zout, size_t zout_cols,
double (*activation_derivative)(double));//TODO
#endif
|
