summaryrefslogtreecommitdiff
path: root/src/spires_interface.h
blob: 006b9d741c9ae1d6cdb4aa5bf20f3151bfd642a2 (plain)
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
#ifndef SPIRES_INTERFACE_H
#define SPIRES_INTERFACE_H

#include <spires.h>
#include <stddef.h>

typedef struct {
	size_t num_samples; // number of time steps
	size_t num_features;
	double *states;
} Reservoir_State_Matrix;

typedef struct {
	double g_min;
	double g_max;
	double alpha;
	double max_abs_weight;
} conductance_mapping;

int collect_reservoir_states(spires_reservoir *reservoir,
			     const double *input_series, size_t series_length,
			     Reservoir_State_Matrix *result);

int train_reservoir(spires_reservoir *reservoir, double *input_series,
		    double *taret_series, size_t series_length, double lambda);

int convert_weights_to_resistances(const spires_reservoir *reservoir,
				   size_t num_neurons, size_t num_outputs,
				   double r_on, double r_off,
				   double **resistances_out,
				   conductance_mapping *mapping);

void free_reservoir_state_matrix(Reservoir_State_Matrix *matrix);

#endif