Giu/analistica
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ex-6 compute EMD of the convolution

Browse Source
This commit is contained in:
Michele Guerini Rocco 2020-05-18 00:32:32 +02:00
parent b9775b3602
commit 9303f4f379
2 changed files with 50 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
ex-6/dist-plot.py
View File

@ -8,21 +8,29 @@ rcParams['font.size'] = 12
n = int(input()) n = int(input())
a, b, f = loadtxt(sys.stdin, unpack=True) a, b, f = loadtxt(sys.stdin, unpack=True)
subplot(121) subplot(131)
title('FFT') title('FFT')
xlabel('EDM distance') xlabel('EDM')
ylabel('counts') ylabel('counts')
hist(a[:n], insert(b[:n], 0, a[0]), weights=f[:n], hist(a[:n], insert(b[:n], 0, a[0]), weights=f[:n],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b') histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0)) ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
subplot(122) subplot(132)
title('RL') title('RL')
xlabel('EDM distance') xlabel('EDM')
ylabel('counts') ylabel('counts')
hist(a[n:], insert(b[n:], 0, a[n]), weights=f[n:], hist(a[n:2*n], insert(b[n:2*n], 0, a[n]), weights=f[n:2*n],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b') histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0)) ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
subplot(133)
title('convolution')
xlabel('EDM')
ylabel('counts')
hist(a[2*n:], insert(b[2*n:], 0, a[2*n]), weights=f[2*n:],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
tight_layout() tight_layout()
show() show()

65
ex-6/test.c
View File

@ -21,14 +21,6 @@ struct options {
double noise; double noise;
}; };
/* A pair of `double`s */
typedef struct {
double a;
double b;
} pair_t;
int show_help(char **argv) { int show_help(char **argv) {
fprintf(stderr, "Usage: %s -[hexbsrn]\n", argv[0]); fprintf(stderr, "Usage: %s -[hexbsrn]\n", argv[0]);
fprintf(stderr, " -h\t\tShow this message.\n"); fprintf(stderr, " -h\t\tShow this message.\n");
@ -61,7 +53,7 @@ int show_help(char **argv) {
* The function returns a pair of the FFT and RL * The function returns a pair of the FFT and RL
* distances, in this order. * distances, in this order.
*/ */
pair_t experiment( double *experiment(
struct options opts, struct options opts,
gsl_rng *r, gsl_rng *r,
gsl_histogram *kernel) { gsl_histogram *kernel) {
@ -103,9 +95,10 @@ pair_t experiment(
* histogram and store add each one to the respective * histogram and store add each one to the respective
* distance histogram. * distance histogram.
*/ */
pair_t dist; static double dist[3];
dist.a = emd_between(hist, fft_clean); dist[0] = emd_between(hist, fft_clean);
dist.b = emd_between(hist, rl_clean); dist[1] = emd_between(hist, rl_clean);
dist[2] = emd_between(hist, conv);
// free memory // free memory
gsl_histogram_free(hist); gsl_histogram_free(hist);
@ -149,13 +142,15 @@ int main(int argc, char **argv) {
/* Performs experiments and record the deconvolution /* Performs experiments and record the deconvolution
* EMD distances to the original sample. * EMD distances to the original sample.
*/ */
double* fft_dist = calloc(opts.num_exps, sizeof(double)); double* fft_dist = calloc(opts.num_exps, sizeof(double));
double* rl_dist = calloc(opts.num_exps, sizeof(double)); double* rl_dist = calloc(opts.num_exps, sizeof(double));
double* conv_dist = calloc(opts.num_exps, sizeof(double));
for (size_t i = 0; i < opts.num_exps; i++) { for (size_t i = 0; i < opts.num_exps; i++) {
pair_t dist = experiment(opts, r, kernel); double *dist = experiment(opts, r, kernel);
fft_dist[i] = dist.a; fft_dist[i] = dist[0];
rl_dist[i] = dist.b; rl_dist[i] = dist[1];
conv_dist[i] = dist[2];
} }
/* Compute some statistics of the EMD */ /* Compute some statistics of the EMD */
@ -175,18 +170,21 @@ int main(int argc, char **argv) {
double conv_skew = gsl_stats_skew_m_sd(conv_dist, 1, opts.num_exps, conv_mean, conv_stdev); double conv_skew = gsl_stats_skew_m_sd(conv_dist, 1, opts.num_exps, conv_mean, conv_stdev);
double conv_min, conv_max; gsl_stats_minmax(&conv_min, &conv_max, conv_dist, 1, opts.num_exps); double conv_min, conv_max; gsl_stats_minmax(&conv_min, &conv_max, conv_dist, 1, opts.num_exps);
/* Create EMD distance histograms. /* Create EDM distance histograms.
* Since the distance depends wildly on the noise we can't * Since the distance depends wildly on the noise we can't
* set a fixed range and therefore use the above values. * set a fixed range and therefore use the above values.
*/ */
gsl_histogram *fft_hist = gsl_histogram_alloc(sqrt(opts.num_exps)); gsl_histogram *fft_hist = gsl_histogram_alloc(sqrt(opts.num_exps));
gsl_histogram *rl_hist = gsl_histogram_alloc(sqrt(opts.num_exps)); gsl_histogram *rl_hist = gsl_histogram_alloc(sqrt(opts.num_exps));
gsl_histogram_set_ranges_uniform(fft_hist, fft_min, fft_max); gsl_histogram *conv_hist = gsl_histogram_alloc(sqrt(opts.num_exps));
gsl_histogram_set_ranges_uniform(rl_hist, rl_min, rl_max); gsl_histogram_set_ranges_uniform( fft_hist, fft_min, fft_max);
gsl_histogram_set_ranges_uniform( rl_hist, rl_min, rl_max);
gsl_histogram_set_ranges_uniform(conv_hist, conv_min, conv_max);
for (size_t i = 0; i < opts.num_exps; i++) { for (size_t i = 0; i < opts.num_exps; i++) {
gsl_histogram_increment(fft_hist, fft_dist[i]); gsl_histogram_increment( fft_hist, fft_dist[i]);
gsl_histogram_increment( rl_hist, rl_dist[i]); gsl_histogram_increment( rl_hist, rl_dist[i]);
gsl_histogram_increment(conv_hist, conv_dist[i]);
} }
// print results to stderr // print results to stderr
@ -199,20 +197,27 @@ int main(int argc, char **argv) {
fprintf(stderr, "- mean: %.2e\n" fprintf(stderr, "- mean: %.2e\n"
"- stdev: %.1e\n" "- stdev: %.1e\n"
"- skew: %.2f\n", rl_mean, rl_stdev, rl_skew); "- skew: %.2f\n", rl_mean, rl_stdev, rl_skew);
fprintf(stderr, "\n## convolution\n");
fprintf(stderr, "- mean: %.2e\n"
"- stdev: %.1e\n"
"- skew: %.2f\n", conv_mean, conv_stdev, conv_skew);
// print histograms to stdout // print histograms to stdout
fprintf(stderr, "\n# EMD histogram\n"); fprintf(stderr, "\n# EMD histogram\n");
fprintf(stdout, "%d\n", (int)sqrt(opts.num_exps)); fprintf(stdout, "%d\n", (int)sqrt(opts.num_exps));
gsl_histogram_fprintf(stdout, fft_hist, "%g", "%g"); gsl_histogram_fprintf(stdout, fft_hist, "%.10g", "%.10g");
gsl_histogram_fprintf(stdout, rl_hist, "%g", "%g"); gsl_histogram_fprintf(stdout, rl_hist, "%.10g", "%.10g");
gsl_histogram_fprintf(stdout, conv_hist, "%.10g", "%.10g");
// free memory // free memory
gsl_rng_free(r); gsl_rng_free(r);
gsl_histogram_free(kernel); gsl_histogram_free(kernel);
gsl_histogram_free(fft_hist); gsl_histogram_free( fft_hist);
gsl_histogram_free(rl_hist); gsl_histogram_free( rl_hist);
free(fft_dist); gsl_histogram_free(conv_hist);
free(rl_dist); free( fft_dist);
free( rl_dist);
free(conv_dist);
return EXIT_SUCCESS; return EXIT_SUCCESS;
} }