Trained software weights with ridge regression

1 files changed, 21 insertions, 2 deletions

diff --git a/src/spires_interface.h b/src/spires_interface.h
index 0a8aa8c..ebf4b76 100644
--- a/src/spires_interface.h
+++ b/src/spires_interface.h
@@ -5,18 +5,37 @@
 #include <spires.h>
 
 typedef struct {
-    size_t num_samples;
+    size_t num_samples; //number of time steps
     size_t num_features;
     double *states;
 } Reservoir_State_Matrix;
 
 int collect_reservoir_states(
-    const spires_reservoir *reservoir,
+    spires_reservoir *reservoir,
     const double *input_series,
     size_t series_length,
     Reservoir_State_Matrix *result
 );
 
+int map_signed_weights_to_resistance(
+        const double *weights,
+        size_t num_neurons,
+        size_t num_outputs,
+        double resistance_on,
+        double resistance_off,
+        double *resistances,
+        double *conductance_offset,
+        double *conductance_scale
+);
+
+int train_reservoir(
+        spires_reservoir *reservoir,
+        double *input_series,
+        double *taret_series,
+        size_t series_length,
+        double lambda
+);
+
 void free_reservoir_state_matrix(
         Reservoir_State_Matrix *matrix
 );