1 files changed, 17 insertions, 0 deletions
diff --git a/include/spiral.h b/include/spiral.h
new file mode 100644
index 0000000..5ed315e
--- /dev/null
+++ b/include/spiral.h
@@ -0,0 +1,17 @@
+#pragma once
+
+#include "utils.h"
+#include "dynmlp.h"
+
+typedef struct {
+    double **z0;
+    double **target;
+    int num_samples;
+} Dataset;
+
+Dataset generate_spiral_dataset(int num_samples, double t0, double t1,
+                                double noise_std, RNG *r);
+void    dataset_free(Dataset *ds);
+double  evaluate(const DynMLP *net, const double *theta,
+                 const Dataset *ds, double t0, double t1,
+                 double atol, double rtol);