include/spiral.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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);