Cleaned up main loop

Setting up for a public api

1 files changed, 6 insertions, 4 deletions

diff --git a/src/benchmark.h b/src/benchmark.h
index cd79d50..4da73d7 100644
--- a/src/benchmark.h
+++ b/src/benchmark.h
@@ -13,15 +13,17 @@ typedef struct {
 
 int run_benchmark(const spires_reservoir_config *config,
 		  spires_reservoir *reservoir,
-		  Reservoir_State_Matrix *state_matrix,
-		  const double *target_outputs, const char *model_path,
-		  const char *subcircuit_name);
+		  Reservoir_State_Matrix *state_matrix, const char *model_path,
+		  const char *subcircuit_name, double *predictions_out);
+
+double calculate_MSE(const double *expected, const double *predicted,
+		     const size_t num_steps, const size_t num_outputs);
 
 int plot_raster(const Reservoir_State_Matrix *matrix, size_t neurons_to_plot,
 		double spike_threshold);
 
 int plot_reservoir_predictions(const double *expected, const double *predicted,
 			       size_t num_samples, size_t num_outputs,
-			       size_t output_to_plot);
+			       size_t output_to_plot, const char *model_path);
 
 #endif