From 43c2a6f3f9cc442eb9c0e69157d11348c80d4149 Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 12 Aug 2026 12:21:01 -0700 Subject: Cleaned up main loop Setting up for a public api --- src/benchmark.h | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'src/benchmark.h') 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 -- cgit v1.2.3