summaryrefslogtreecommitdiff
path: root/src/spires_interface.h
blob: f3552b90d8b4d4273e583dc3a3d85c28d9e1d516 (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
36
37
38
#ifndef SPIRES_INTERFACE_H
#define SPIRES_INTERFACE_H

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

typedef struct {
	size_t num_samples;
	size_t num_features;
	/* Row-major continuous reservoir states: [sample][feature]. */
	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);

/* Train the SPIRES-owned readout using its configured state representation. */
int train_reservoir(spires_reservoir *reservoir, double *input_series,
		    double *target_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);

/* Release storage allocated by collect_reservoir_states(). */
void free_reservoir_state_matrix(Reservoir_State_Matrix *matrix);

#endif