neural-network/dataset.h at main · gab-dev-7/neural-network · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#ifndef DATASET_H
#define DATASET_H

typedef struct {
    double** inputs;
    double** targets;
    int num_samples;
    int input_size;
    int output_size;
} Dataset;

// dataset generation functions
Dataset* create_sine_wave_dataset(int num_samples);
Dataset* create_xor_dataset();
Dataset* create_circle_dataset(int num_samples);
Dataset* create_circle_dataset_enhanced(int num_samples, double boundary_ratio);

// utility functions
void split_dataset(Dataset* dataset, double train_ratio,
                   Dataset** train_set, Dataset** test_set);
void split_dataset_three_way(Dataset* dataset, double train_ratio, double val_ratio,
                             Dataset** train_set, Dataset** val_set, Dataset** test_set);
void shuffle_dataset(Dataset* dataset);
void free_dataset(Dataset* dataset);

#endif