Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 1 | #!/usr/local/bin/perl |
| 2 | use Inline C; |
| 3 | use Mojolicious::Lite; |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 4 | use Mojo::JSON qw(decode_json encode_json to_json); |
Marc Kupietz | 247500f | 2015-10-09 11:29:01 +0200 | [diff] [blame] | 5 | use Encode qw(decode encode); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 6 | use Getopt::Std; |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 7 | use Mojo::Server::Daemon; |
Marc Kupietz | d422739 | 2016-03-01 16:45:12 +0100 | [diff] [blame] | 8 | plugin 'Log::Access'; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 9 | |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 10 | our $opt_i = 0; # latin1-input? |
| 11 | our $opt_l = undef; |
| 12 | our $opt_p = 5676; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 13 | our $opt_n = undef; |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 14 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 15 | getopt('il:p:n:'); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 16 | |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 17 | # -cbow 1 -size 200 -window 8 -negative 25 -hs 0 -sample 1e-4 -threads 40 -binary 1 -iter 15 |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 18 | if(!$ARGV[0]) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 19 | init_net("vectors15.bin", $opt_n, ($opt_i? 1 : 0)); |
Marc Kupietz | 2cb667e | 2016-03-10 09:44:12 +0100 | [diff] [blame] | 20 | } else { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 21 | init_net($ARGV[0], $opt_n, ($opt_i? 1 : 0)); |
Marc Kupietz | 2cb667e | 2016-03-10 09:44:12 +0100 | [diff] [blame] | 22 | } |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 23 | |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 24 | my $daemon = Mojo::Server::Daemon->new( |
| 25 | app => app, |
| 26 | listen => ['http://'.($opt_l ? $opt_l : '*').":$opt_p"] |
| 27 | ); |
| 28 | |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 29 | get '/' => sub { |
| 30 | my $c = shift; |
| 31 | my $word=$c->param('word'); |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 32 | my $no_nbs=$c->param('n') || 100; |
| 33 | my $no_iterations=$c->param('N') || 2000; |
Marc Kupietz | d422739 | 2016-03-01 16:45:12 +0100 | [diff] [blame] | 34 | my $perplexity=$c->param('perplexity') || 20; |
Marc Kupietz | c4d62f8 | 2016-03-01 11:04:24 +0100 | [diff] [blame] | 35 | my $epsilon=$c->param('epsilon') || 5; |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 36 | my $som=$c->param('som') || 0; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 37 | my $res; |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 38 | my @lists; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 39 | my @collocations; |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 40 | if(defined($word) && $word !~ /^\s*$/) { |
| 41 | $c->inactivity_timeout(300); |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 42 | $word =~ s/\s+/ /g; |
| 43 | for my $w (split(' *\| *', $word)) { |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 44 | $c->app->log->debug('Looking for neighbours of '.$w); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 45 | if($opt_i) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 46 | $res = get_neighbours(encode("iso-8859-1", $w), $no_nbs); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 47 | } else { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 48 | $res = get_neighbours($w, $no_nbs); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 49 | } |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 50 | push(@lists, $res->{paradigmatic}); |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 51 | } |
Marc Kupietz | 247500f | 2015-10-09 11:29:01 +0200 | [diff] [blame] | 52 | } |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 53 | $word =~ s/ *\| */ | /g; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 54 | $c->render(template=>"index", word=>$word, no_nbs=>$no_nbs, no_iterations => $no_iterations, epsilon=> $epsilon, perplexity=> $perplexity, show_som=>$som, lists=> \@lists, collocators=> $res->{syntagmatic}); |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 55 | }; |
| 56 | |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 57 | $daemon->run; # app->start; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 58 | |
| 59 | exit; |
| 60 | |
| 61 | __END__ |
| 62 | |
| 63 | __C__ |
| 64 | #include <stdio.h> |
| 65 | #include <string.h> |
| 66 | #include <math.h> |
| 67 | #include <malloc.h> |
| 68 | #include <stdlib.h> //strlen |
Marc Kupietz | f080976 | 2016-02-26 10:13:47 +0100 | [diff] [blame] | 69 | #include <sys/mman.h> |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 70 | #include <pthread.h> |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 71 | |
| 72 | #define max_size 2000 |
| 73 | #define max_w 50 |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 74 | #define MAX_NEIGHBOURS 1000 |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 75 | #define MAX_WORDS -1 |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 76 | #define MAX_THREADS 100 |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 77 | #define MAX_CC 50 |
| 78 | #define EXP_TABLE_SIZE 1000 |
| 79 | #define MAX_EXP 6 |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 80 | |
| 81 | //the thread function |
| 82 | void *connection_handler(void *); |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 83 | |
| 84 | typedef struct { |
| 85 | long long *index; |
| 86 | float *dist; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 87 | float *norm; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 88 | long long *pos; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 89 | unsigned int length; |
| 90 | } knn; |
| 91 | |
| 92 | |
| 93 | typedef struct { |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 94 | long long wordi[MAX_NEIGHBOURS]; |
| 95 | char sep[MAX_NEIGHBOURS]; |
| 96 | int length; |
| 97 | } wordlist; |
| 98 | |
| 99 | typedef struct { |
| 100 | wordlist *wl; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 101 | char *token; |
| 102 | int N; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 103 | long from; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 104 | unsigned long upto; |
| 105 | } knnpars; |
| 106 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 107 | float *M, *syn1neg_window, *expTable; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 108 | char *vocab; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 109 | |
Marc Kupietz | 82b0267 | 2016-02-26 12:32:25 +0100 | [diff] [blame] | 110 | long long words, size; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 111 | int num_threads=20; |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 112 | int latin_enc=0; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 113 | int window; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 114 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 115 | int init_net(char *file_name, char *net_name, int latin) { |
Marc Kupietz | 67c2028 | 2016-02-26 09:42:00 +0100 | [diff] [blame] | 116 | FILE *f, *binvecs, *binwords; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 117 | int binwords_fd, binvecs_fd, net_fd, i; |
Marc Kupietz | 82b0267 | 2016-02-26 12:32:25 +0100 | [diff] [blame] | 118 | long long a, b, c, d, cn; |
| 119 | float len; |
| 120 | |
Marc Kupietz | 67c2028 | 2016-02-26 09:42:00 +0100 | [diff] [blame] | 121 | char binvecs_fname[256], binwords_fname[256]; |
| 122 | strcpy(binwords_fname, file_name); |
| 123 | strcat(binwords_fname, ".words"); |
| 124 | strcpy(binvecs_fname, file_name); |
| 125 | strcat(binvecs_fname, ".vecs"); |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 126 | |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 127 | latin_enc = latin; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 128 | f = fopen(file_name, "rb"); |
| 129 | if (f == NULL) { |
| 130 | printf("Input file %s not found\n", file_name); |
| 131 | return -1; |
| 132 | } |
| 133 | fscanf(f, "%lld", &words); |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 134 | if(MAX_WORDS > 0 && words > MAX_WORDS) words = MAX_WORDS; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 135 | fscanf(f, "%lld", &size); |
Marc Kupietz | 2cb667e | 2016-03-10 09:44:12 +0100 | [diff] [blame] | 136 | if( (binvecs_fd = open(binvecs_fname, O_RDONLY)) < 0 || (binwords_fd = open(binwords_fname, O_RDONLY)) < 0) { |
| 137 | printf("Converting %s to memory mappable structures\n", file_name); |
Marc Kupietz | f080976 | 2016-02-26 10:13:47 +0100 | [diff] [blame] | 138 | vocab = (char *)malloc((long long)words * max_w * sizeof(char)); |
| 139 | M = (float *)malloc((long long)words * (long long)size * sizeof(float)); |
| 140 | if (M == NULL) { |
| 141 | printf("Cannot allocate memory: %lld MB %lld %lld\n", (long long)words * size * sizeof(float) / 1048576, words, size); |
| 142 | return -1; |
| 143 | } |
Marc Kupietz | 67c2028 | 2016-02-26 09:42:00 +0100 | [diff] [blame] | 144 | for (b = 0; b < words; b++) { |
| 145 | a = 0; |
| 146 | while (1) { |
| 147 | vocab[b * max_w + a] = fgetc(f); |
| 148 | if (feof(f) || (vocab[b * max_w + a] == ' ')) break; |
| 149 | if ((a < max_w) && (vocab[b * max_w + a] != '\n')) a++; |
| 150 | } |
| 151 | vocab[b * max_w + a] = 0; |
| 152 | fread(&M[b * size], sizeof(float), size, f); |
| 153 | len = 0; |
| 154 | for (a = 0; a < size; a++) len += M[a + b * size] * M[a + b * size]; |
| 155 | len = sqrt(len); |
| 156 | for (a = 0; a < size; a++) M[a + b * size] /= len; |
| 157 | } |
| 158 | if( (binvecs = fopen(binvecs_fname, "wb")) != NULL && (binwords = fopen(binwords_fname, "wb")) != NULL) { |
| 159 | fwrite(M, sizeof(float), (long long)words * (long long)size, binvecs); |
| 160 | fclose(binvecs); |
| 161 | fwrite(vocab, sizeof(char), (long long)words * max_w, binwords); |
| 162 | fclose(binwords); |
| 163 | } |
Marc Kupietz | 2cb667e | 2016-03-10 09:44:12 +0100 | [diff] [blame] | 164 | } |
| 165 | if( (binvecs_fd = open(binvecs_fname, O_RDONLY)) >= 0 && (binwords_fd = open(binwords_fname, O_RDONLY)) >= 0) { |
| 166 | M = mmap(0, sizeof(float) * (long long)words * (long long)size, PROT_READ, MAP_SHARED, binvecs_fd, 0); |
| 167 | vocab = mmap(0, sizeof(char) * (long long)words * max_w, PROT_READ, MAP_SHARED, binwords_fd, 0); |
| 168 | if (M == MAP_FAILED || vocab == MAP_FAILED) { |
| 169 | close(binvecs_fd); |
| 170 | close(binwords_fd); |
| 171 | fprintf(stderr, "Cannot mmap %s or %s\n", binwords_fname, binvecs_fname); |
| 172 | exit(-1); |
| 173 | } |
| 174 | } else { |
| 175 | fprintf(stderr, "Cannot open %s or %s\n", binwords_fname, binvecs_fname); |
| 176 | exit(-1); |
Marc Kupietz | 67c2028 | 2016-02-26 09:42:00 +0100 | [diff] [blame] | 177 | } |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 178 | fclose(f); |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 179 | |
| 180 | if(net_name) { |
| 181 | if( (net_fd = open(net_name, O_RDONLY)) >= 0) { |
| 182 | window = (lseek(net_fd, 0, SEEK_END) - sizeof(float) * words * size) / words / size / sizeof(float) / 2; |
| 183 | // lseek(net_fd, sizeof(float) * words * size, SEEK_SET); |
| 184 | munmap(M, sizeof(float) * words * size); |
| 185 | M = mmap(0, sizeof(float) * words * size + sizeof(float) * 2 * window * size * words, PROT_READ, MAP_SHARED, net_fd, 0); |
| 186 | if (M == MAP_FAILED) { |
| 187 | close(net_fd); |
| 188 | fprintf(stderr, "Cannot mmap %s\n", net_name); |
| 189 | exit(-1); |
| 190 | } |
| 191 | syn1neg_window = M + words * size; |
| 192 | } else { |
| 193 | fprintf(stderr, "Cannot open %s\n", net_name); |
| 194 | exit(-1); |
| 195 | } |
| 196 | fprintf(stderr, "Successfully memmaped %s. Determined window size: %d\n", net_name, window); |
| 197 | } |
| 198 | |
| 199 | expTable = (float *) malloc((EXP_TABLE_SIZE + 1) * sizeof(float)); |
| 200 | for (i = 0; i < EXP_TABLE_SIZE; i++) { |
| 201 | expTable[i] = exp((i / (float) EXP_TABLE_SIZE * 2 - 1) * MAX_EXP); // Precompute the exp() table |
| 202 | expTable[i] = expTable[i] / (expTable[i] + 1); // Precompute f(x) = x / (x + 1) |
| 203 | } |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 204 | return 0; |
| 205 | } |
| 206 | |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 207 | knn *getCollocators(int cc, int N) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 208 | knn *nbs = NULL; |
| 209 | long window_layer_size = size * window * 2; |
| 210 | long a, b, c, d, e, window_offset, target, max_target=0, maxmax_target; |
| 211 | float f, max_f, maxmax_f; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 212 | float *target_sums, *bestf, *bestn, worstbest, wpos_sum; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 213 | long long *besti, *bestp; |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 214 | |
| 215 | if(cc == -1) |
| 216 | return NULL; |
| 217 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 218 | a = posix_memalign((void **) &target_sums, 128, words * sizeof(float)); |
| 219 | besti = malloc(N * sizeof(long long)); |
| 220 | bestp = malloc(N * sizeof(long long)); |
| 221 | bestf = malloc(N * sizeof(float)); |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 222 | bestn = malloc(N * sizeof(float)); |
| 223 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 224 | for (b = 0; b < words; b++) |
| 225 | target_sums[b]=0; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 226 | for (b = 0; b < N; b++) { |
| 227 | bestn[b] = 1; |
| 228 | bestf[b] = -1; |
| 229 | } |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 230 | worstbest = -1; |
| 231 | d = cc; |
| 232 | maxmax_f = -1; |
| 233 | maxmax_target = 0; |
| 234 | |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 235 | besti[0]=d; |
| 236 | bestf[0]=1.0; |
| 237 | bestp[0]=0; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 238 | for (a = window * 2 + 1; a >=0; a--) { |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 239 | wpos_sum = 0; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 240 | printf("window pos: %ld\n", a); |
| 241 | if (a != window) { |
| 242 | max_f = -1; |
| 243 | window_offset = a * size; |
| 244 | if (a > window) |
| 245 | window_offset -= size; |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 246 | for(target = 0; target < words / 2; target ++) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 247 | if(target == d) |
| 248 | continue; |
| 249 | f = 0; |
| 250 | for (c = 0; c < size; c++) |
| 251 | f += M[d* size + c] * syn1neg_window[target * window_layer_size + window_offset + c]; |
| 252 | if (f < -MAX_EXP) |
| 253 | continue; |
| 254 | else if (f > MAX_EXP) |
| 255 | continue; |
| 256 | else |
| 257 | f = expTable[(int) ((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 258 | wpos_sum += f; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 259 | if(f > max_f) { |
| 260 | max_f = f; |
| 261 | max_target = target; |
| 262 | } |
| 263 | target_sums[target] += (1-target_sums[target]) * f; |
| 264 | if(f > worstbest) { |
| 265 | for (b = 0; b < N; b++) { |
| 266 | if (f > bestf[b]) { |
| 267 | for (e = N - 1; e > b; e--) { |
| 268 | bestf[e] = bestf[e - 1]; |
| 269 | besti[e] = besti[e - 1]; |
| 270 | bestp[e] = bestp[e - 1]; |
| 271 | } |
| 272 | bestf[b] = f; |
| 273 | besti[b] = target; |
| 274 | bestp[b] = window-a; |
| 275 | break; |
| 276 | } |
| 277 | } |
| 278 | worstbest = bestf[N-1]; |
| 279 | } |
| 280 | } |
| 281 | printf("%d %.2f\n", max_target, max_f); |
| 282 | printf("%s (%.2f) ", &vocab[max_target * max_w], max_f); |
| 283 | if(max_f > maxmax_f) { |
| 284 | maxmax_f = max_f; |
| 285 | maxmax_target = max_target; |
| 286 | } |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 287 | for (b = 0; b < N; b++) |
| 288 | if(bestp[b] == window-a) |
| 289 | bestn[b] = bestf[b] / wpos_sum; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 290 | } else { |
| 291 | printf("\x1b[1m%s\x1b[0m ", &vocab[d*max_w]); |
| 292 | } |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 293 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 294 | } |
| 295 | max_f = -1; |
| 296 | for (b = 0; b < words; b++) { |
| 297 | if(target_sums[b] > max_f) { |
| 298 | max_f = target_sums[b]; |
| 299 | max_target = b; |
| 300 | } |
| 301 | } |
| 302 | printf(" -- max sum: %s (%.2f), max resp.: \x1b[1m%s\x1b[0m (%.2f)\n", |
| 303 | &vocab[max_target * max_w], max_f, |
| 304 | &vocab[maxmax_target * max_w], maxmax_f); |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 305 | for(b=0; b<N; b++) |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 306 | printf("%-32s %.2f %d\n", &vocab[besti[b]*max_w], bestf[b], bestp[b]); |
| 307 | printf("\n"); |
| 308 | free(target_sums); |
| 309 | nbs = malloc(sizeof(knn)); |
| 310 | nbs->index = besti; |
| 311 | nbs->dist = bestf; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 312 | nbs->norm = bestn; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 313 | nbs->pos = bestp; |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 314 | nbs->length = N; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 315 | return(nbs); |
| 316 | } |
| 317 | |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 318 | wordlist *getTargetWords(char *st1) { |
| 319 | wordlist *wl = malloc(sizeof(wordlist)); |
| 320 | char st[100][max_size], sep[100]; |
| 321 | long a, b=0, c=0, cn=0; |
| 322 | |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 323 | while (1) { |
| 324 | st[cn][b] = st1[c]; |
| 325 | b++; |
| 326 | c++; |
| 327 | st[cn][b] = 0; |
| 328 | if (st1[c] == 0) break; |
Marc Kupietz | 95aa1c0 | 2016-03-15 09:40:43 +0100 | [diff] [blame] | 329 | if (st1[c] == ' ' || st1[c] == '-') { |
| 330 | sep[cn++] = st1[c]; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 331 | b = 0; |
| 332 | c++; |
| 333 | } |
| 334 | } |
| 335 | cn++; |
| 336 | for (a = 0; a < cn; a++) { |
Marc Kupietz | 34a3ee9 | 2016-02-27 22:43:16 +0100 | [diff] [blame] | 337 | for (b = 0; b < words; b++) if (!strcmp(&vocab[b * max_w], st[a])) break; |
| 338 | if (b == words) b = -1; |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 339 | wl->wordi[a] = b; |
| 340 | fprintf(stderr, "Word: \"%s\" Position in vocabulary: %lld\n", st[a], wl->wordi[a]); |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 341 | if (b == -1) { |
Marc Kupietz | e8da306 | 2016-02-25 08:37:53 +0100 | [diff] [blame] | 342 | fprintf(stderr, "Out of dictionary word!\n"); |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 343 | cn--; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 344 | break; |
| 345 | } |
| 346 | } |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 347 | wl->length=cn; |
| 348 | return(wl); |
| 349 | } |
| 350 | |
| 351 | void *_get_neighbours(knnpars *pars) { |
| 352 | char *st1 = pars->token; |
| 353 | int N = pars->N; |
| 354 | long from = pars -> from; |
| 355 | unsigned long upto = pars -> upto; |
| 356 | char file_name[max_size], st[100][max_size], *sep; |
| 357 | float dist, len, *bestd, vec[max_size]; |
| 358 | long long a, b, c, d, cn, *bi, *besti; |
| 359 | char ch; |
| 360 | knn *nbs = NULL; |
| 361 | wordlist *wl = pars->wl; |
| 362 | |
| 363 | besti = malloc(N * sizeof(long long)); |
| 364 | bestd = malloc(N * sizeof(float)); |
| 365 | |
| 366 | float worstbest=-1; |
| 367 | |
| 368 | for (a = 0; a < N; a++) bestd[a] = 0; |
| 369 | a = 0; |
| 370 | bi = wl->wordi; |
| 371 | cn = wl->length; |
| 372 | sep = wl->sep; |
| 373 | b = bi[0]; |
| 374 | c = 0; |
| 375 | |
| 376 | if(from < 0) { |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 377 | nbs = getCollocators(b, pars->N); |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 378 | pthread_exit(nbs); |
| 379 | } |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 380 | if (b == -1) { |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 381 | N = 0; |
| 382 | goto end; |
| 383 | } |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 384 | for (a = 0; a < size; a++) vec[a] = 0; |
| 385 | for (b = 0; b < cn; b++) { |
| 386 | if (bi[b] == -1) continue; |
Marc Kupietz | 95aa1c0 | 2016-03-15 09:40:43 +0100 | [diff] [blame] | 387 | if(b>0 && sep[b-1] == '-') |
| 388 | for (a = 0; a < size; a++) vec[a] -= M[a + bi[b] * size]; |
| 389 | else |
| 390 | for (a = 0; a < size; a++) vec[a] += M[a + bi[b] * size]; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 391 | } |
| 392 | len = 0; |
| 393 | for (a = 0; a < size; a++) len += vec[a] * vec[a]; |
| 394 | len = sqrt(len); |
| 395 | for (a = 0; a < size; a++) vec[a] /= len; |
| 396 | for (a = 0; a < N; a++) bestd[a] = -1; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 397 | for (c = from; c < upto; c++) { |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 398 | a = 0; |
Marc Kupietz | 34020dc | 2016-02-25 08:44:19 +0100 | [diff] [blame] | 399 | // do not skip taget word |
Marc Kupietz | e8da306 | 2016-02-25 08:37:53 +0100 | [diff] [blame] | 400 | // for (b = 0; b < cn; b++) if (bi[b] == c) a = 1; |
| 401 | // if (a == 1) continue; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 402 | dist = 0; |
| 403 | for (a = 0; a < size; a++) dist += vec[a] * M[a + c * size]; |
Marc Kupietz | be1b9fc | 2016-02-26 10:34:30 +0100 | [diff] [blame] | 404 | if(dist > worstbest) { |
| 405 | for (a = 0; a < N; a++) { |
| 406 | if (dist > bestd[a]) { |
| 407 | for (d = N - 1; d > a; d--) { |
| 408 | bestd[d] = bestd[d - 1]; |
| 409 | besti[d] = besti[d - 1]; |
| 410 | } |
| 411 | bestd[a] = dist; |
| 412 | besti[a] = c; |
| 413 | break; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 414 | } |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 415 | } |
Marc Kupietz | be1b9fc | 2016-02-26 10:34:30 +0100 | [diff] [blame] | 416 | worstbest = bestd[N-1]; |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 417 | } |
| 418 | } |
Marc Kupietz | 34020dc | 2016-02-25 08:44:19 +0100 | [diff] [blame] | 419 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 420 | nbs = malloc(sizeof(knn)); |
| 421 | nbs->index = besti; |
| 422 | nbs->dist = bestd; |
| 423 | nbs->length = N; |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 424 | end: |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 425 | pthread_exit(nbs); |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 426 | } |
| 427 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 428 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 429 | SV *get_neighbours(char *st1, int N) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 430 | HV *result = newHV(); |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 431 | float bestd[MAX_NEIGHBOURS], bestn[MAX_NEIGHBOURS], vec[max_size]; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 432 | long long besti[MAX_NEIGHBOURS], bestp[MAX_NEIGHBOURS], a, b, c, d, slice; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 433 | char *bestw[MAX_NEIGHBOURS]; |
| 434 | knn *nbs[MAX_THREADS]; |
| 435 | knnpars pars[MAX_THREADS]; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 436 | pthread_t *pt = (pthread_t *)malloc((num_threads+1) * sizeof(pthread_t)); |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 437 | wordlist *wl; |
| 438 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 439 | if(N>MAX_NEIGHBOURS) N=MAX_NEIGHBOURS; |
| 440 | |
| 441 | slice = words / num_threads; |
| 442 | |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 443 | wl = getTargetWords(st1); |
| 444 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 445 | a = num_threads; |
| 446 | pars[a].token = st1; |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 447 | pars[a].wl = wl; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 448 | pars[a].N = N; |
| 449 | pars[a].from = -1; |
| 450 | pthread_create(&pt[a], NULL, _get_neighbours, (void *) &pars[a]); |
| 451 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 452 | for(a=0; a < num_threads; a++) { |
| 453 | pars[a].token = st1; |
Marc Kupietz | 48c2968 | 2016-03-19 11:30:43 +0100 | [diff] [blame] | 454 | pars[a].wl = wl; |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 455 | pars[a].N = N; |
| 456 | pars[a].from = a*slice; |
| 457 | pars[a].upto = ((a+1)*slice > words? words:(a+1)*slice); |
| 458 | pthread_create(&pt[a], NULL, _get_neighbours, (void *) &pars[a]); |
| 459 | } |
| 460 | for (a = 0; a < num_threads; a++) pthread_join(pt[a], &nbs[a]); |
| 461 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 462 | pthread_join(pt[a], &nbs[a]); |
| 463 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 464 | if(!nbs[0]) |
| 465 | goto end; |
| 466 | |
| 467 | for(b=0; b < N; b++) { |
| 468 | besti[b] = nbs[0]->index[b]; |
| 469 | bestd[b] = nbs[0]->dist[b]; |
| 470 | } |
| 471 | |
| 472 | for(a=1; a < num_threads; a++) { |
| 473 | for(b=0; b < N; b++) { |
| 474 | for(c=0; c < N; c++) { |
| 475 | if(nbs[a]->dist[b] > bestd[c]) { |
| 476 | for(d=N-1; d>c; d--) { |
| 477 | bestd[d] = bestd[d-1]; |
| 478 | besti[d] = besti[d-1]; |
| 479 | } |
| 480 | besti[c] = nbs[a]->index[b]; |
| 481 | bestd[c] = nbs[a]->dist[b]; |
| 482 | break; |
| 483 | } |
| 484 | } |
| 485 | } |
| 486 | } |
| 487 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 488 | |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 489 | if(nbs) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 490 | AV* array = newAV(); |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 491 | for (a = 0; a < N; a++) { |
| 492 | bestw[a] = (char *)malloc(max_size * sizeof(char)); |
| 493 | } |
| 494 | for (a = 0; a < N; a++) { |
| 495 | strcpy(bestw[a], &vocab[besti[a] * max_w]); |
| 496 | HV* hash = newHV(); |
Marc Kupietz | a5b9015 | 2016-03-15 17:39:19 +0100 | [diff] [blame] | 497 | SV* word = newSVpvf(bestw[a], 0); |
| 498 | if(latin_enc == 0) SvUTF8_on(word); |
| 499 | hv_store(hash, "word", strlen("word"), word , 0); |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 500 | hv_store(hash, "dist", strlen("dist"), newSVnv(bestd[a]), 0); |
| 501 | hv_store(hash, "rank", strlen("rank"), newSVuv(besti[a]), 0); |
| 502 | AV *vector = newAV(); |
| 503 | for (b = 0; b < size; b++) { |
| 504 | av_push(vector, newSVnv(M[b + besti[a] * size])); |
| 505 | } |
| 506 | hv_store(hash, "vector", strlen("vector"), newRV_noinc((SV*)vector), 0); |
| 507 | av_push(array, newRV_noinc((SV*)hash)); |
| 508 | } |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 509 | hv_store(result, "paradigmatic", strlen("paradigmatic"), newRV_noinc((SV*)array), 0); |
| 510 | |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 511 | for(b=0; b < nbs[num_threads]->length; b++) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 512 | besti[b] = nbs[num_threads]->index[b]; |
| 513 | bestd[b] = nbs[num_threads]->dist[b]; |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 514 | bestn[b] = nbs[num_threads]->norm[b]; |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 515 | bestp[b] = nbs[num_threads]->pos[b]; |
| 516 | } |
| 517 | array = newAV(); |
Marc Kupietz | da9db6f | 2016-03-19 20:36:20 +0100 | [diff] [blame] | 518 | for (a = 0; a < nbs[num_threads]->length; a++) { |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 519 | strcpy(bestw[a], &vocab[besti[a] * max_w]); |
| 520 | HV* hash = newHV(); |
| 521 | SV* word = newSVpvf(bestw[a], 0); |
| 522 | if(latin_enc == 0) SvUTF8_on(word); |
| 523 | hv_store(hash, "word", strlen("word"), word , 0); |
| 524 | hv_store(hash, "dist", strlen("dist"), newSVnv(bestd[a]), 0); |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 525 | hv_store(hash, "norm", strlen("norm"), newSVnv(bestn[a]), 0); |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 526 | hv_store(hash, "pos", strlen("pos"), newSVnv(bestp[a]), 0); |
| 527 | av_push(array, newRV_noinc((SV*)hash)); |
| 528 | } |
| 529 | hv_store(result, "syntagmatic", strlen("syntagmatic"), newRV_noinc((SV*)array), 0); |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 530 | } |
| 531 | end: |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 532 | return newRV_noinc((SV*)result); |
Marc Kupietz | 000ad86 | 2016-02-26 14:59:12 +0100 | [diff] [blame] | 533 | } |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 534 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 535 | |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 536 | __DATA__ |
| 537 | |
| 538 | @@ index.html.ep |
| 539 | <!DOCTYPE html> |
| 540 | <html> |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 541 | <head> |
| 542 | <title>DeReKo-Word-Vector-Distances</title> |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 543 | <link rel="stylesheet" href="//code.jquery.com/ui/1.11.4/themes/smoothness/jquery-ui.css"> |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 544 | <script src="http://code.jquery.com/jquery-latest.min.js"></script> |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 545 | <script src="//code.jquery.com/ui/1.11.4/jquery-ui.js"></script> |
| 546 | <script> |
| 547 | $(function() { |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 548 | $( document ).tooltip({ |
| 549 | content: function() { |
| 550 | return $(this).attr('title'); |
| 551 | }} |
| 552 | ) |
| 553 | }) |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 554 | </script> |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 555 | <script src="//d3js.org/d3.v3.min.js" charset="utf-8"></script> |
| 556 | <script src="http://klinux10/word2vec/tsne.js"></script> |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 557 | <script src="http://klinux10/word2vec/som.js"></script> |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 558 | <script src="http://klinux10/word2vec/labeler.js"></script> |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 559 | <style> |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 560 | body, input { |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 561 | font-family: Arial, sans-serif; |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 562 | font-size: 11pt; |
| 563 | } |
| 564 | |
| 565 | .ui-tooltip-content { |
| 566 | font-size: 9pt; |
| 567 | colour: #222222; |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 568 | } |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 569 | |
| 570 | svg > .ui-tooltip-content { |
Marc Kupietz | b102936 | 2016-02-27 21:38:55 +0100 | [diff] [blame] | 571 | font-size: 8pt; |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 572 | colour: #222222; |
| 573 | } |
| 574 | |
Marc Kupietz | 6b2975c | 2016-03-18 21:59:33 +0100 | [diff] [blame] | 575 | #collocators { |
| 576 | margin-bottom: 15px; |
| 577 | } |
| 578 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 579 | #wrapper { |
| 580 | width: 100%; |
| 581 | // border: 1px solid red; |
| 582 | overflow: hidden; /* will contain if #first is longer than #second */ |
| 583 | } |
| 584 | #first { |
Marc Kupietz | b102936 | 2016-02-27 21:38:55 +0100 | [diff] [blame] | 585 | margin-right: 20px; |
| 586 | float: left; |
| 587 | // border: 1px solid green; |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 588 | } |
| 589 | #second { |
| 590 | border: 1px solid #333; |
| 591 | overflow: hidden; /* if you don't want #second to wrap below #first */ |
| 592 | } |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 593 | #som2 svg { |
| 594 | border: 1px solid #333; |
| 595 | } |
| 596 | |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 597 | #cost { |
Marc Kupietz | b102936 | 2016-02-27 21:38:55 +0100 | [diff] [blame] | 598 | font-size: 8pt; |
| 599 | color: #222222; |
| 600 | margin-top: 4px; |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 601 | margin-bottom: 12px; |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 602 | } |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 603 | |
Marc Kupietz | 6c1ca44 | 2016-03-03 09:35:18 +0100 | [diff] [blame] | 604 | #sominfo1, #sominfo { |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 605 | font-size: 8pt; |
| 606 | color: #222222; |
| 607 | margin-top: 0px; |
| 608 | } |
| 609 | |
Marc Kupietz | 6c1ca44 | 2016-03-03 09:35:18 +0100 | [diff] [blame] | 610 | #somcolor1, #somcolor2, #somcolor3 { |
| 611 | display: inline-block; |
| 612 | height: 10px; |
| 613 | width: 10px; |
| 614 | } |
| 615 | |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 616 | #third { |
| 617 | border: 1px solid #333; |
| 618 | } |
| 619 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 620 | </style> |
| 621 | <script> |
| 622 | |
Marc Kupietz | c4d62f8 | 2016-03-01 11:04:24 +0100 | [diff] [blame] | 623 | var opt = {epsilon: <%= $epsilon %>, perplexity: <%= $perplexity %>}, |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 624 | mapWidth = 800, // width map |
| 625 | mapHeight = 800, |
| 626 | jitterRadius = 7; |
| 627 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 628 | var T = new tsnejs.tSNE(opt); // create a tSNE instance |
| 629 | |
| 630 | var Y; |
| 631 | |
| 632 | var data; |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 633 | var labeler; |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 634 | |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 635 | |
| 636 | function applyJitter() { |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 637 | svg.selectAll('.tsnet') |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 638 | .data(labels) |
| 639 | .transition() |
| 640 | .duration(50) |
| 641 | .attr("transform", function(d, i) { |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 642 | T.Y[i][0] = (d.x - mapWidth/2 - tx)/ss/20; |
| 643 | T.Y[i][1] = (d.y - mapHeight/2 - ty)/ss/20; |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 644 | return "translate(" + |
| 645 | (d.x) + "," + |
| 646 | (d.y) + ")"; |
| 647 | }); |
| 648 | } |
| 649 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 650 | function updateEmbedding() { |
| 651 | var Y = T.getSolution(); |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 652 | svg.selectAll('.tsnet') |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 653 | .data(data.words) |
| 654 | .attr("transform", function(d, i) { |
| 655 | return "translate(" + |
| 656 | ((Y[i][0]*20*ss + tx) + mapWidth/2) + "," + |
| 657 | ((Y[i][1]*20*ss + ty) + mapHeight/2) + ")"; }); |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 658 | } |
| 659 | |
| 660 | var svg; |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 661 | var labels = []; |
| 662 | var anchor_array = []; |
| 663 | var text; |
| 664 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 665 | function drawEmbedding() { |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 666 | $("#embed").empty(); |
| 667 | var div = d3.select("#embed"); |
| 668 | |
| 669 | // get min and max in each column of Y |
| 670 | var Y = T.Y; |
| 671 | |
| 672 | svg = div.append("svg") // svg is global |
| 673 | .attr("width", mapWidth) |
| 674 | .attr("height", mapHeight); |
| 675 | |
| 676 | var g = svg.selectAll(".b") |
| 677 | .data(data.words) |
| 678 | .enter().append("g") |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 679 | .attr("class", "tsnet"); |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 680 | |
| 681 | g.append("a") |
| 682 | .attr("xlink:href", function(word) {return "/?word="+word;}) |
| 683 | .attr("title", function(d, i) { |
Marc Kupietz | e50c816 | 2016-03-01 10:24:43 +0100 | [diff] [blame] | 684 | return "rank: "+i +" "+"freq. rank: "+data.ranks[i].toString().replace(/\B(?=(\d{3})+(?!\d))/g, ","); |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 685 | }) |
Marc Kupietz | a350bce | 2016-02-25 09:34:25 +0100 | [diff] [blame] | 686 | .append("text") |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 687 | .attr("text-anchor", "top") |
| 688 | .attr("font-size", 12) |
| 689 | .attr("fill", function(d) { |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 690 | if(data.target.indexOf(" "+d+" ") >= 0) { |
Marc Kupietz | a350bce | 2016-02-25 09:34:25 +0100 | [diff] [blame] | 691 | return "red"; |
| 692 | } else { |
| 693 | return "#333" |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 694 | } |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 695 | }) |
| 696 | .text(function(d) { return d; }); |
| 697 | |
| 698 | var zoomListener = d3.behavior.zoom() |
| 699 | .scaleExtent([0.1, 10]) |
| 700 | .center([0,0]) |
| 701 | .on("zoom", zoomHandler); |
| 702 | zoomListener(svg); |
| 703 | } |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 704 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 705 | var tx=0, ty=0; |
| 706 | var ss=1; |
| 707 | var iter_id=-1; |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 708 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 709 | function zoomHandler() { |
| 710 | tx = d3.event.translate[0]; |
| 711 | ty = d3.event.translate[1]; |
| 712 | ss = d3.event.scale; |
| 713 | updateEmbedding(); |
| 714 | } |
| 715 | |
| 716 | var stepnum = 0; |
| 717 | |
| 718 | function stopStep() { |
| 719 | clearInterval(iter_id); |
| 720 | text = svg.selectAll("text"); |
| 721 | |
| 722 | // jitter function needs different data and co-ordinate representation |
| 723 | labels = d3.range(data.words.length).map(function(i) { |
| 724 | var x = (T.Y[i][0]*20*ss + tx) + mapWidth/2; |
| 725 | var y = (T.Y[i][1]*20*ss + ty) + mapHeight/2; |
| 726 | anchor_array.push({x: x, y: y, r: jitterRadius}); |
| 727 | return { |
| 728 | x: x, |
| 729 | y: y, |
| 730 | name: data.words[i] |
| 731 | }; |
| 732 | }); |
| 733 | |
| 734 | // get the actual label bounding boxes for the jitter function |
| 735 | var index = 0; |
| 736 | text.each(function() { |
| 737 | labels[index].width = this.getBBox().width; |
| 738 | labels[index].height = this.getBBox().height; |
| 739 | index += 1; |
| 740 | }); |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 741 | |
| 742 | |
| 743 | // setTimeout(updateEmbedding, 1); |
| 744 | // setTimeout( |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 745 | labeler = d3.labeler() |
| 746 | .label(labels) |
| 747 | .anchor(anchor_array) |
| 748 | .width(mapWidth) |
| 749 | .height(mapHeight) |
| 750 | .update(applyJitter); |
| 751 | // .start(1000); |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 752 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 753 | iter_id = setInterval(jitterStep, 1); |
| 754 | } |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 755 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 756 | var jitter_i=0; |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 757 | |
| 758 | function jitterStep() { |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 759 | if(jitter_i++ > 100) { |
| 760 | clearInterval(iter_id); |
| 761 | } else { |
| 762 | labeler.start2(10); |
| 763 | applyJitter(); |
| 764 | } |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 765 | } |
Marc Kupietz | b102936 | 2016-02-27 21:38:55 +0100 | [diff] [blame] | 766 | |
| 767 | var last_cost=1000; |
| 768 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 769 | function step() { |
| 770 | var i = T.iter; |
| 771 | |
| 772 | if(i > <%= $no_iterations %>) { |
| 773 | stopStep(); |
| 774 | } else { |
| 775 | var cost = Math.round(T.step() * 100000) / 100000; // do a few steps |
| 776 | $("#cost").html("tsne iteration " + i + ", cost: " + cost.toFixed(5)); |
| 777 | if(i % 250 == 0 && cost >= last_cost) { |
| 778 | stopStep(); |
| 779 | } else { |
| 780 | last_cost = cost; |
| 781 | updateEmbedding(); |
| 782 | } |
| 783 | } |
| 784 | } |
Marc Kupietz | c5990da | 2016-02-26 08:47:12 +0100 | [diff] [blame] | 785 | |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 786 | function showMap(j) { |
| 787 | data=j; |
| 788 | T.iter=0; |
| 789 | T.initDataRaw(data.vecs); // init embedding |
| 790 | drawEmbedding(); // draw initial embedding |
| 791 | |
| 792 | if(iter_id >= 0) { |
| 793 | clearInterval(iter_id); |
| 794 | } |
| 795 | //T.debugGrad(); |
| 796 | iter_id = setInterval(step, 1); |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 797 | if(<%= $show_som %>) { |
| 798 | makeSOM(j, <%= $no_iterations %>); |
| 799 | } |
Marc Kupietz | 9fca173 | 2016-02-29 09:07:04 +0100 | [diff] [blame] | 800 | } |
Marc Kupietz | a350bce | 2016-02-25 09:34:25 +0100 | [diff] [blame] | 801 | |
Marc Kupietz | c489336 | 2016-02-25 08:04:46 +0100 | [diff] [blame] | 802 | </script> |
| 803 | </head> |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 804 | <body> |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 805 | <form action="<%=url_for('/')->to_abs%>" method="GET"> |
Marc Kupietz | 44bee3c | 2016-02-25 16:26:29 +0100 | [diff] [blame] | 806 | word(s): |
| 807 | <input type="text" name="word" size="20" value="<%= $word %>" title="When looking for multiple words use spaces as separators to search around the average vector and | as separator to get the neighbours for each word."> |
| 808 | max. neighbours: <input type="text" size="8" name="n" value="<%= $no_nbs %>"> |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 809 | max. iterations: <input type="text" name="N" size="8" value="<%= $no_iterations %>"> |
| 810 | SOM <input type="checkbox" name="som" value="1" <%= ($show_som ? "checked" : "") %>> |
| 811 | <span> </span><input type="submit" value="Show"> |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 812 | </form> |
| 813 | <br> |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 814 | % if($lists) { |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 815 | <div id="wrapper"> |
| 816 | <table id="first"> |
| 817 | <tr> |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 818 | <th align="right">#</th><th align="right">cos</th><th align="left">paradigmatic</th><th title="Position in winodw around target word. Absolute value can be too low because of sub-sampling frequent words.">@</th><th align="right" title=""Responsivenes" of the collocator at the relative position @. Approximation of the probability that the combination of the target word and the collocator at the relative position @ come from the corpus.">resp.</th><th title="Probability of the collocator at window location @."align="right">p(c<sub><small>@</small></sub>)</th><th align="left">syntagmatic</th> |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 819 | </tr> |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 820 | % my $j=0; my @words; my @vecs; my @ranks; for my $list (@$lists) { |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 821 | % my $i=0; for my $item (@$list) { |
| 822 | % my $c = (@$collocators)[$i]; |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 823 | % if(!grep{$_ eq $item->{word}} @words) { |
| 824 | % push @vecs, $item->{vector}; |
| 825 | % push @words, $item->{word}; |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 826 | % push @ranks, $item->{rank}; |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 827 | % } |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 828 | <tr> |
| 829 | <td align="right"> |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 830 | <%= ++$i %>. |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 831 | </td> |
| 832 | <td align="right"> |
| 833 | <%= sprintf("%.3f", $item->{dist}) %> |
| 834 | </td> |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 835 | <td> |
| 836 | <a title="freq. rank: <%= $item->{rank} %>" href="/?word=<%= $item->{word} %>"> |
| 837 | <%= $item->{word} %> |
| 838 | </a> |
| 839 | </td> |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 840 | <td align="right"> |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 841 | <%= $c->{pos} %>: |
| 842 | </td> |
| 843 | <td align="right"> |
| 844 | <%= sprintf("%.3f", $c->{dist}) %> |
| 845 | </td> |
Marc Kupietz | b864ccf | 2016-03-21 22:40:03 +0100 | [diff] [blame^] | 846 | <td align="right"> |
| 847 | <%= sprintf("%.3e", $c->{norm}) %> |
| 848 | </td> |
Marc Kupietz | d564258 | 2016-03-19 22:23:13 +0100 | [diff] [blame] | 849 | <td align="left"> |
| 850 | <a href="/?word=<%= $c->{word} %>"> |
| 851 | <%= $c->{word} %> |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 852 | </td> |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 853 | </tr> |
| 854 | % } |
Marc Kupietz | 7b2cbeb | 2016-02-25 11:22:00 +0100 | [diff] [blame] | 855 | % } |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 856 | </table> |
| 857 | <script> |
| 858 | % use Mojo::ByteStream 'b'; |
| 859 | $(window).load(function() { |
Marc Kupietz | 5f78067 | 2016-02-25 17:15:54 +0100 | [diff] [blame] | 860 | showMap(<%= b(Mojo::JSON::to_json({target => " $word ", words => \@words, vecs => \@vecs, ranks => \@ranks})); %>); |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 861 | }); |
| 862 | </script> |
| 863 | % } |
| 864 | <div id="second" style="width:800px; height:800px; font-family: arial;"> |
| 865 | <div id="embed"> |
| 866 | </div> |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 867 | </div> |
Marc Kupietz | b102936 | 2016-02-27 21:38:55 +0100 | [diff] [blame] | 868 | <div id="cost"></div> |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 869 | % if($show_som) { |
| 870 | <div id="som2"> |
| 871 | </div> |
Marc Kupietz | 6c1ca44 | 2016-03-03 09:35:18 +0100 | [diff] [blame] | 872 | <div id="sominfo1"><span id="somcolor1"> </span> <span id="somword1"> </span> <span id="somcolor2"> </span> <span id="somword2"> </span> <span id="somcolor3"> </span></div> |
Marc Kupietz | d7aea72 | 2016-03-02 11:59:12 +0100 | [diff] [blame] | 873 | <div id="sominfo">SOM iteration <span id="iterations">0</span></div> |
| 874 | % } |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 875 | </div> |
| 876 | <p> |
| 877 | Word vector model based on DeReKo-2015-II. Trained with <a href="https://code.google.com/p/word2vec/">word2vec</a> using the following parameters:</p> |
Marc Kupietz | 247500f | 2015-10-09 11:29:01 +0200 | [diff] [blame] | 878 | <pre> |
Marc Kupietz | 7bc85fd | 2016-02-24 11:42:41 +0100 | [diff] [blame] | 879 | -cbow 1 -size 300 -window 7 -negative 5 -hs 0 -sample 1e-5 -threads 44 -binary 1 -iter 5 |
Marc Kupietz | 4aa6217 | 2016-02-25 10:39:27 +0100 | [diff] [blame] | 880 | </pre> |
| 881 | </p> |
| 882 | </body> |
Marc Kupietz | dc22b98 | 2015-10-09 09:19:34 +0200 | [diff] [blame] | 883 | </html> |
| 884 | |