blob: 1173a183da75b0189e13ddd3b050bc390ed0bbe8 [file] [log] [blame]
Marc Kupietzdc22b982015-10-09 09:19:34 +02001#!/usr/local/bin/perl
2use Inline C;
Marc Kupietza2e64502016-04-27 09:53:51 +02003use Inline C => Config => CLEAN_AFTER_BUILD => 0, ccflags => $Config{ccflags}." -O4";
Marc Kupietza5f60042017-05-04 10:38:12 +02004#use Inline C => Config => BUILD_NOISY => 1, CFLAGS => $Config{cflags}." -O4 -mtune k9";
5#use Inline C => Config => CLEAN_AFTER_BUILD => 0, ccflags => $Config{ccflags}." -Ofast -march k8 -mtune k8 ";
Marc Kupietzdc22b982015-10-09 09:19:34 +02006use Mojolicious::Lite;
Marc Kupietzc4893362016-02-25 08:04:46 +01007use Mojo::JSON qw(decode_json encode_json to_json);
Marc Kupietz247500f2015-10-09 11:29:01 +02008use Encode qw(decode encode);
Marc Kupietza5b90152016-03-15 17:39:19 +01009use Getopt::Std;
Marc Kupietz7bc85fd2016-02-24 11:42:41 +010010use Mojo::Server::Daemon;
Marc Kupietzffef9302017-11-07 15:58:01 +010011use Cwd;
12app->static->paths->[0] = getcwd;
13
Marc Kupietzd4227392016-03-01 16:45:12 +010014plugin 'Log::Access';
Marc Kupietzb3422c12017-07-04 14:12:11 +020015plugin "RequestBase";
Marc Kupietzdc22b982015-10-09 09:19:34 +020016
Marc Kupietza5b90152016-03-15 17:39:19 +010017our $opt_i = 0; # latin1-input?
18our $opt_l = undef;
19our $opt_p = 5676;
Marc Kupietza2e64502016-04-27 09:53:51 +020020our $opt_m;
Marc Kupietz6ed81872016-04-27 14:04:04 +020021our $opt_M;
Marc Kupietz43ee87e2016-04-25 10:50:08 +020022our $opt_n = '';
23our $opt_d;
Marc Kupietz5c3887d2016-04-28 08:53:35 +020024our $opt_G;
Marc Kupietza5b90152016-03-15 17:39:19 +010025
Marc Kupietz6ed81872016-04-27 14:04:04 +020026my %marked;
Marc Kupietz793413b2016-04-02 21:48:57 +020027my $training_args="";
Marc Kupietza2e64502016-04-27 09:53:51 +020028my $mergedEnd=0;
Marc Kupietz15987412017-11-07 15:56:58 +010029my %cache;
Marc Kupietz793413b2016-04-02 21:48:57 +020030
Marc Kupietza5f60042017-05-04 10:38:12 +020031getopts('d:Gil:p:m:n:M:');
Marc Kupietz6ed81872016-04-27 14:04:04 +020032
33if($opt_M) {
Marc Kupietzed930212016-04-27 15:42:38 +020034 open my $handle, '<:encoding(UTF-8)', $opt_M
35 or die "Can't open '$opt_M' for reading: $!";
36 while(<$handle>) {
Marc Kupietz6ed81872016-04-27 14:04:04 +020037 foreach my $mw (split /\s+/) {
38 $marked{$mw}=1
39 }
40 }
Marc Kupietzed930212016-04-27 15:42:38 +020041 close($handle);
Marc Kupietz6ed81872016-04-27 14:04:04 +020042}
Marc Kupietza5b90152016-03-15 17:39:19 +010043
Marc Kupietz7bc85fd2016-02-24 11:42:41 +010044# -cbow 1 -size 200 -window 8 -negative 25 -hs 0 -sample 1e-4 -threads 40 -binary 1 -iter 15
Marc Kupietza5b90152016-03-15 17:39:19 +010045if(!$ARGV[0]) {
Marc Kupietz6b2975c2016-03-18 21:59:33 +010046 init_net("vectors15.bin", $opt_n, ($opt_i? 1 : 0));
Marc Kupietz2cb667e2016-03-10 09:44:12 +010047} else {
Marc Kupietz6b2975c2016-03-18 21:59:33 +010048 init_net($ARGV[0], $opt_n, ($opt_i? 1 : 0));
Marc Kupietz793413b2016-04-02 21:48:57 +020049 if(open(FILE, "$ARGV[0].args")) {
50 $training_args = <FILE>;
51 }
52 close(FILE);
Marc Kupietz2cb667e2016-03-10 09:44:12 +010053}
Marc Kupietzdc22b982015-10-09 09:19:34 +020054
Marc Kupietza2e64502016-04-27 09:53:51 +020055if($opt_m) {
56 $mergedEnd = mergeVectors($opt_m);
57}
58
Marc Kupietz6ed81872016-04-27 14:04:04 +020059
Marc Kupietz43ee87e2016-04-25 10:50:08 +020060if($opt_d) { # -d: dump vecs and exit
61 dump_vecs($opt_d);
62 exit;
63}
64
Marc Kupietza5b90152016-03-15 17:39:19 +010065my $daemon = Mojo::Server::Daemon->new(
66 app => app,
67 listen => ['http://'.($opt_l ? $opt_l : '*').":$opt_p"]
68);
69
Marc Kupietz5c3887d2016-04-28 08:53:35 +020070if($opt_G) {
71 print "Filtering garbage\n";
72 filter_garbage();
73}
74
Marc Kupietz554aff52017-11-09 14:42:09 +010075get '*/js/*' => sub {
Marc Kupietzffef9302017-11-07 15:58:01 +010076 my $c = shift;
77 my $url = $c->req->url;
78 $url =~ s@/derekovecs@@g;
79 $c->app->log->info("GET: " . $url);
80 $c->reply->static($url);
81};
82
Marc Kupietzdc22b982015-10-09 09:19:34 +020083get '/' => sub {
84 my $c = shift;
Marc Kupietza5f60042017-05-04 10:38:12 +020085 $c->app->log->info("get: ".$c->req->url->to_abs);
Marc Kupietzdc22b982015-10-09 09:19:34 +020086 my $word=$c->param('word');
Marc Kupietz44bee3c2016-02-25 16:26:29 +010087 my $no_nbs=$c->param('n') || 100;
88 my $no_iterations=$c->param('N') || 2000;
Marc Kupietzd4227392016-03-01 16:45:12 +010089 my $perplexity=$c->param('perplexity') || 20;
Marc Kupietzc4d62f82016-03-01 11:04:24 +010090 my $epsilon=$c->param('epsilon') || 5;
Marc Kupietzd7aea722016-03-02 11:59:12 +010091 my $som=$c->param('som') || 0;
Marc Kupietza2e64502016-04-27 09:53:51 +020092 my $searchBaseVocabFirst=$c->param('sbf') || 0;
Marc Kupietz6d9a6782016-03-23 17:25:25 +010093 my $sort=$c->param('sort') || 0;
Marc Kupietzb613b052016-04-28 14:11:59 +020094 my $json=$c->param('json') || 0;
Marc Kupietz2c79c5e2017-11-09 16:18:40 +010095 my $cutoff=$c->param('cutoff') || 1000000;
Marc Kupietz6b2975c2016-03-18 21:59:33 +010096 my $res;
Marc Kupietz7b2cbeb2016-02-25 11:22:00 +010097 my @lists;
Marc Kupietz6b2975c2016-03-18 21:59:33 +010098 my @collocations;
Marc Kupietz7bc85fd2016-02-24 11:42:41 +010099 if(defined($word) && $word !~ /^\s*$/) {
100 $c->inactivity_timeout(300);
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100101 $word =~ s/\s+/ /g;
102 for my $w (split(' *\| *', $word)) {
Marc Kupietze2840922017-11-09 16:20:29 +0100103 if ($cache{$w.$cutoff.$no_nbs.$sort}) {
Marc Kupietz15987412017-11-07 15:56:58 +0100104 $c->app->log->info("Getting $w results from cache");
Marc Kupietze2840922017-11-09 16:20:29 +0100105 $res = $cache{$w.$cutoff.$no_nbs.$sort}
Marc Kupietza5b90152016-03-15 17:39:19 +0100106 } else {
Marc Kupietz15987412017-11-07 15:56:58 +0100107 $c->app->log->info('Looking for neighbours of '.$w);
108 if($opt_i) {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100109 $res = get_neighbours(encode("iso-8859-1", $w), $no_nbs, $sort, $searchBaseVocabFirst, $cutoff);
Marc Kupietz15987412017-11-07 15:56:58 +0100110 } else {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100111 $res = get_neighbours($w, $no_nbs, $sort, $searchBaseVocabFirst, $cutoff);
Marc Kupietz15987412017-11-07 15:56:58 +0100112 }
113 $cache{$w} = $res;
Marc Kupietza5b90152016-03-15 17:39:19 +0100114 }
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100115 push(@lists, $res->{paradigmatic});
Marc Kupietz15987412017-11-07 15:56:58 +0100116 }
117 }
Marc Kupietz000ad862016-02-26 14:59:12 +0100118 $word =~ s/ *\| */ | /g;
Marc Kupietzb613b052016-04-28 14:11:59 +0200119 if($json) {
120 return $c->render(json => {word => $word, list => \@lists, collocators=>$res->{syntagmatic}});
121 } else {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100122 $c->render(template=>"index", word=>$word, cutoff=>$cutoff, no_nbs=>$no_nbs, no_iterations => $no_iterations, epsilon=> $epsilon, perplexity=> $perplexity, show_som=>$som, searchBaseVocabFirst=>$searchBaseVocabFirst, sort=>$sort, training_args=>$training_args, mergedEnd=> $mergedEnd, marked=>\%marked, lists=> \@lists, collocators=> $res->{syntagmatic});
Marc Kupietzb613b052016-04-28 14:11:59 +0200123 }
Marc Kupietzdc22b982015-10-09 09:19:34 +0200124};
125
Marc Kupietza5b90152016-03-15 17:39:19 +0100126$daemon->run; # app->start;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200127
128exit;
129
130__END__
131
132__C__
133#include <stdio.h>
134#include <string.h>
135#include <math.h>
136#include <malloc.h>
137#include <stdlib.h> //strlen
Marc Kupietzf0809762016-02-26 10:13:47 +0100138#include <sys/mman.h>
Marc Kupietz000ad862016-02-26 14:59:12 +0100139#include <pthread.h>
Marc Kupietzdc22b982015-10-09 09:19:34 +0200140
141#define max_size 2000
142#define max_w 50
Marc Kupietz7bc85fd2016-02-24 11:42:41 +0100143#define MAX_NEIGHBOURS 1000
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100144#define MAX_WORDS -1
Marc Kupietz000ad862016-02-26 14:59:12 +0100145#define MAX_THREADS 100
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100146#define MAX_CC 50
147#define EXP_TABLE_SIZE 1000
148#define MAX_EXP 6
Marc Kupietz271e2a42016-03-22 11:37:43 +0100149#define MIN_RESP 0.50
Marc Kupietzdc22b982015-10-09 09:19:34 +0200150
151//the thread function
152void *connection_handler(void *);
Marc Kupietz000ad862016-02-26 14:59:12 +0100153
154typedef struct {
155 long long *index;
156 float *dist;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100157 float *norm;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100158 long long *pos;
Marc Kupietz80abb442016-03-23 21:04:08 +0100159 int length;
Marc Kupietz000ad862016-02-26 14:59:12 +0100160} knn;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100161
Marc Kupietz000ad862016-02-26 14:59:12 +0100162typedef struct {
Marc Kupietz48c29682016-03-19 11:30:43 +0100163 long long wordi[MAX_NEIGHBOURS];
164 char sep[MAX_NEIGHBOURS];
165 int length;
166} wordlist;
167
168typedef struct {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100169 long cutoff;
Marc Kupietz48c29682016-03-19 11:30:43 +0100170 wordlist *wl;
Marc Kupietz000ad862016-02-26 14:59:12 +0100171 char *token;
172 int N;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100173 long from;
Marc Kupietz000ad862016-02-26 14:59:12 +0100174 unsigned long upto;
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100175 float *target_sums;
Marc Kupietz000ad862016-02-26 14:59:12 +0100176} knnpars;
177
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200178float *M, *M2=0L, *syn1neg_window, *expTable;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200179char *vocab;
Marc Kupietz5c3887d2016-04-28 08:53:35 +0200180char *garbage = NULL;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100181
Marc Kupietza2e64502016-04-27 09:53:51 +0200182long long words, size, merged_end;
Marc Kupietza5f60042017-05-04 10:38:12 +0200183long long merge_words = 0;
Marc Kupietz000ad862016-02-26 14:59:12 +0100184int num_threads=20;
Marc Kupietza5b90152016-03-15 17:39:19 +0100185int latin_enc=0;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100186int window;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200187
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100188int init_net(char *file_name, char *net_name, int latin) {
Marc Kupietz67c20282016-02-26 09:42:00 +0100189 FILE *f, *binvecs, *binwords;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100190 int binwords_fd, binvecs_fd, net_fd, i;
Marc Kupietz82b02672016-02-26 12:32:25 +0100191 long long a, b, c, d, cn;
192 float len;
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200193 double val;
Marc Kupietz82b02672016-02-26 12:32:25 +0100194
Marc Kupietz67c20282016-02-26 09:42:00 +0100195 char binvecs_fname[256], binwords_fname[256];
196 strcpy(binwords_fname, file_name);
197 strcat(binwords_fname, ".words");
198 strcpy(binvecs_fname, file_name);
199 strcat(binvecs_fname, ".vecs");
Marc Kupietzdc22b982015-10-09 09:19:34 +0200200
Marc Kupietza5b90152016-03-15 17:39:19 +0100201 latin_enc = latin;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200202 f = fopen(file_name, "rb");
203 if (f == NULL) {
204 printf("Input file %s not found\n", file_name);
205 return -1;
206 }
207 fscanf(f, "%lld", &words);
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100208 if(MAX_WORDS > 0 && words > MAX_WORDS) words = MAX_WORDS;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200209 fscanf(f, "%lld", &size);
Marc Kupietz2cb667e2016-03-10 09:44:12 +0100210 if( (binvecs_fd = open(binvecs_fname, O_RDONLY)) < 0 || (binwords_fd = open(binwords_fname, O_RDONLY)) < 0) {
211 printf("Converting %s to memory mappable structures\n", file_name);
Marc Kupietzf0809762016-02-26 10:13:47 +0100212 vocab = (char *)malloc((long long)words * max_w * sizeof(char));
213 M = (float *)malloc((long long)words * (long long)size * sizeof(float));
214 if (M == NULL) {
215 printf("Cannot allocate memory: %lld MB %lld %lld\n", (long long)words * size * sizeof(float) / 1048576, words, size);
216 return -1;
217 }
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200218 if(strstr(file_name, ".txt")) {
219 for (b = 0; b < words; b++) {
220 a = 0;
221 while (1) {
222 vocab[b * max_w + a] = fgetc(f);
223 if (feof(f) || (vocab[b * max_w + a] == ' ')) break;
224 if ((a < max_w) && (vocab[b * max_w + a] != '\n')) a++;
225 }
226 vocab[b * max_w + a] = 0;
227 len = 0;
228 for (a = 0; a < size; a++) {
229 fscanf(f, "%lf", &val);
230 M[a + b * size] = val;
231 len += val * val;
232 }
233 len = sqrt(len);
234 for (a = 0; a < size; a++) M[a + b * size] /= len;
235 }
236 } else {
237 for (b = 0; b < words; b++) {
238 a = 0;
239 while (1) {
240 vocab[b * max_w + a] = fgetc(f);
241 if (feof(f) || (vocab[b * max_w + a] == ' ')) break;
242 if ((a < max_w) && (vocab[b * max_w + a] != '\n')) a++;
243 }
244 vocab[b * max_w + a] = 0;
245 fread(&M[b * size], sizeof(float), size, f);
246 len = 0;
247 for (a = 0; a < size; a++) len += M[a + b * size] * M[a + b * size];
248 len = sqrt(len);
249 for (a = 0; a < size; a++) M[a + b * size] /= len;
250 }
251 }
Marc Kupietz67c20282016-02-26 09:42:00 +0100252 if( (binvecs = fopen(binvecs_fname, "wb")) != NULL && (binwords = fopen(binwords_fname, "wb")) != NULL) {
253 fwrite(M, sizeof(float), (long long)words * (long long)size, binvecs);
254 fclose(binvecs);
255 fwrite(vocab, sizeof(char), (long long)words * max_w, binwords);
256 fclose(binwords);
257 }
Marc Kupietz2cb667e2016-03-10 09:44:12 +0100258 }
259 if( (binvecs_fd = open(binvecs_fname, O_RDONLY)) >= 0 && (binwords_fd = open(binwords_fname, O_RDONLY)) >= 0) {
260 M = mmap(0, sizeof(float) * (long long)words * (long long)size, PROT_READ, MAP_SHARED, binvecs_fd, 0);
261 vocab = mmap(0, sizeof(char) * (long long)words * max_w, PROT_READ, MAP_SHARED, binwords_fd, 0);
262 if (M == MAP_FAILED || vocab == MAP_FAILED) {
263 close(binvecs_fd);
264 close(binwords_fd);
265 fprintf(stderr, "Cannot mmap %s or %s\n", binwords_fname, binvecs_fname);
266 exit(-1);
267 }
268 } else {
269 fprintf(stderr, "Cannot open %s or %s\n", binwords_fname, binvecs_fname);
270 exit(-1);
Marc Kupietz67c20282016-02-26 09:42:00 +0100271 }
Marc Kupietzdc22b982015-10-09 09:19:34 +0200272 fclose(f);
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100273
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200274 if(net_name && strlen(net_name) > 0) {
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100275 if( (net_fd = open(net_name, O_RDONLY)) >= 0) {
276 window = (lseek(net_fd, 0, SEEK_END) - sizeof(float) * words * size) / words / size / sizeof(float) / 2;
277 // lseek(net_fd, sizeof(float) * words * size, SEEK_SET);
Marc Kupietz10bec2b2016-03-23 09:41:31 +0100278 // munmap(M, sizeof(float) * words * size);
279 M2 = mmap(0, sizeof(float) * words * size + sizeof(float) * 2 * window * size * words, PROT_READ, MAP_SHARED, net_fd, 0);
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200280 if (M2 == MAP_FAILED) {
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100281 close(net_fd);
282 fprintf(stderr, "Cannot mmap %s\n", net_name);
283 exit(-1);
284 }
Marc Kupietz10bec2b2016-03-23 09:41:31 +0100285 syn1neg_window = M2 + words * size;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100286 } else {
287 fprintf(stderr, "Cannot open %s\n", net_name);
288 exit(-1);
289 }
290 fprintf(stderr, "Successfully memmaped %s. Determined window size: %d\n", net_name, window);
291 }
292
293 expTable = (float *) malloc((EXP_TABLE_SIZE + 1) * sizeof(float));
294 for (i = 0; i < EXP_TABLE_SIZE; i++) {
295 expTable[i] = exp((i / (float) EXP_TABLE_SIZE * 2 - 1) * MAX_EXP); // Precompute the exp() table
296 expTable[i] = expTable[i] / (expTable[i] + 1); // Precompute f(x) = x / (x + 1)
297 }
Marc Kupietzdc22b982015-10-09 09:19:34 +0200298 return 0;
299}
300
Marc Kupietza2e64502016-04-27 09:53:51 +0200301long mergeVectors(char *file_name){
302 FILE *f, *binvecs, *binwords;
303 int binwords_fd, binvecs_fd, net_fd, i;
304 long long a, b, c, d, cn;
305 float len;
306 float *merge_vecs;
307 char *merge_vocab;
Marc Kupietza5f60042017-05-04 10:38:12 +0200308 /* long long merge_words, merge_size; */
309 long long merge_size;
Marc Kupietza2e64502016-04-27 09:53:51 +0200310
311 char binvecs_fname[256], binwords_fname[256];
312 strcpy(binwords_fname, file_name);
313 strcat(binwords_fname, ".words");
314 strcpy(binvecs_fname, file_name);
315 strcat(binvecs_fname, ".vecs");
316
317 f = fopen(file_name, "rb");
318 if (f == NULL) {
319 printf("Input file %s not found\n", file_name);
320 exit -1;
321 }
322 fscanf(f, "%lld", &merge_words);
323 fscanf(f, "%lld", &merge_size);
324 if(merge_size != size){
325 fprintf(stderr, "vectors must have the same length\n");
326 exit(-1);
327 }
328 if( (binvecs_fd = open(binvecs_fname, O_RDONLY)) >= 0 && (binwords_fd = open(binwords_fname, O_RDONLY)) >= 0) {
329 merge_vecs = malloc(sizeof(float) * (words + merge_words) * size);
330 merge_vocab = malloc(sizeof(char) * (words + merge_words) * max_w);
331 if (merge_vecs == NULL || merge_vocab == NULL) {
332 close(binvecs_fd);
333 close(binwords_fd);
334 fprintf(stderr, "Cannot reserve memory for %s or %s\n", binwords_fname, binvecs_fname);
335 exit(-1);
336 }
337 read(binvecs_fd, merge_vecs, merge_words * size * sizeof(float));
338 read(binwords_fd, merge_vocab, merge_words * max_w);
339 } else {
340 fprintf(stderr, "Cannot open %s or %s\n", binwords_fname, binvecs_fname);
341 exit(-1);
342 }
343 printf("Successfully reallocated memory\nMerging...\n");
344 fflush(stdout);
345 memcpy(merge_vecs + merge_words * size, M, words * size * sizeof(float));
346 memcpy(merge_vocab + merge_words * max_w, vocab, words * max_w);
347 munmap(M, words * size * sizeof(float));
348 munmap(vocab, words * max_w);
349 M = merge_vecs;
350 vocab = merge_vocab;
351 merged_end = merge_words;
352 words += merge_words;
353 fclose(f);
354 printf("merged_end: %lld, words: %lld\n", merged_end, words);
355 return((long) merged_end);
356}
357
Marc Kupietz5c3887d2016-04-28 08:53:35 +0200358void filter_garbage() {
359 long i;
Marc Kupietzab591a82016-04-28 14:08:49 +0200360 unsigned char *w, previous, c;
Marc Kupietz5c3887d2016-04-28 08:53:35 +0200361 garbage = malloc(words);
362 memset(garbage, 0, words);
363 for (i = 0; i < words; i++) {
364 w = vocab + i * max_w;
365 previous = 0;
Marc Kupietzab591a82016-04-28 14:08:49 +0200366 while((c = *w++) && !garbage[i]) {
367 if( ((c <= 90 && c >= 65) && (previous >= 97 && previous <= 122)) ||
368 (previous == '-' && (c & 32)) ||
Marc Kupietz33e3aa12017-11-09 16:19:38 +0100369 (previous == 0xc2 && (c == 0xa4 || c == 0xb6 )) ||
370 c == '<'
Marc Kupietzab591a82016-04-28 14:08:49 +0200371 ) {
Marc Kupietz5c3887d2016-04-28 08:53:35 +0200372 garbage[i]=1;
373 continue;
374 }
375 previous = c;
376 }
377 }
378 return;
379}
380
Marc Kupietz271e2a42016-03-22 11:37:43 +0100381void *getCollocators(knnpars *pars) {
382 int N = pars->N;
383 int cc = pars->wl->wordi[0];
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100384 knn *nbs = NULL;
385 long window_layer_size = size * window * 2;
386 long a, b, c, d, e, window_offset, target, max_target=0, maxmax_target;
387 float f, max_f, maxmax_f;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100388 float *target_sums, *bestf, *bestn, worstbest, wpos_sum;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100389 long long *besti, *bestp;
Marc Kupietzd5642582016-03-19 22:23:13 +0100390
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200391 if(M2 == NULL || cc == -1)
Marc Kupietzd5642582016-03-19 22:23:13 +0100392 return NULL;
393
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100394 a = posix_memalign((void **) &target_sums, 128, pars->cutoff * sizeof(float));
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100395 besti = malloc(N * sizeof(long long));
396 bestp = malloc(N * sizeof(long long));
397 bestf = malloc(N * sizeof(float));
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100398 bestn = malloc(N * sizeof(float));
399
Marc Kupietz271e2a42016-03-22 11:37:43 +0100400 worstbest = MIN_RESP;
401
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100402 for (b = 0; b < pars->cutoff; b++)
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100403 target_sums[b]=0;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100404 for (b = 0; b < N; b++) {
Marc Kupietz271e2a42016-03-22 11:37:43 +0100405 besti[b] = -1;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100406 bestn[b] = 1;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100407 bestf[b] = worstbest;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100408 }
Marc Kupietz271e2a42016-03-22 11:37:43 +0100409
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100410 d = cc;
411 maxmax_f = -1;
412 maxmax_target = 0;
413
Marc Kupietz271e2a42016-03-22 11:37:43 +0100414 for (a = pars->from; a < pars->upto; a++) {
415 if(a >= window)
416 a++;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100417 wpos_sum = 0;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100418 printf("window pos: %ld\n", a);
419 if (a != window) {
420 max_f = -1;
421 window_offset = a * size;
422 if (a > window)
423 window_offset -= size;
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100424 for(target = 0; target < pars->cutoff; target ++) {
425 if(garbage && garbage[target]) continue;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100426 if(target == d)
427 continue;
428 f = 0;
429 for (c = 0; c < size; c++)
Marc Kupietz10bec2b2016-03-23 09:41:31 +0100430 f += M2[d* size + c] * syn1neg_window[target * window_layer_size + window_offset + c];
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100431 if (f < -MAX_EXP)
432 continue;
433 else if (f > MAX_EXP)
434 continue;
435 else
436 f = expTable[(int) ((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))];
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100437 wpos_sum += f;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100438
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100439 target_sums[target] += f;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100440 if(f > worstbest) {
Marc Kupietz6d9a6782016-03-23 17:25:25 +0100441 for (b = 0; b < N; b++) {
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100442 if (f > bestf[b]) {
Marc Kupietz33679a32016-03-22 08:49:39 +0100443 memmove(bestf + b + 1, bestf + b, (N - b -1) * sizeof(float));
444 memmove(besti + b + 1, besti + b, (N - b -1) * sizeof(long long));
445 memmove(bestp + b + 1, bestp + b, (N - b -1) * sizeof(long long));
446 bestf[b] = f;
447 besti[b] = target;
448 bestp[b] = window-a;
449 break;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100450 }
451 }
Marc Kupietz6d9a6782016-03-23 17:25:25 +0100452 if(b == N - 1)
453 worstbest = bestf[N-1];
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100454 }
455 }
456 printf("%d %.2f\n", max_target, max_f);
457 printf("%s (%.2f) ", &vocab[max_target * max_w], max_f);
458 if(max_f > maxmax_f) {
459 maxmax_f = max_f;
460 maxmax_target = max_target;
461 }
Marc Kupietz33679a32016-03-22 08:49:39 +0100462 for (b = 0; b < N; b++)
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100463 if(bestp[b] == window-a)
464 bestn[b] = bestf[b] / wpos_sum;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100465 } else {
466 printf("\x1b[1m%s\x1b[0m ", &vocab[d*max_w]);
467 }
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100468
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100469 }
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100470 for (b = 0; b < pars->cutoff; b++)
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100471 pars->target_sums[b] += (target_sums[b] / wpos_sum ) / (window * 2);
472 free(target_sums);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100473 for(b=0; b<N && besti[b] >= 0; b++) // THIS LOOP IS NEEDED (b...)
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100474 printf("%s %.2f %d * ", &vocab[besti[b]*max_w], bestf[b], bestp[b]);
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100475 printf("\n");
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100476 nbs = malloc(sizeof(knn));
477 nbs->index = besti;
478 nbs->dist = bestf;
Marc Kupietzb864ccf2016-03-21 22:40:03 +0100479 nbs->norm = bestn;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100480 nbs->pos = bestp;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100481 nbs->length = b-1;
482 pthread_exit(nbs);
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100483}
484
Marc Kupietza2e64502016-04-27 09:53:51 +0200485wordlist *getTargetWords(char *st1, int search_backw) {
Marc Kupietz48c29682016-03-19 11:30:43 +0100486 wordlist *wl = malloc(sizeof(wordlist));
487 char st[100][max_size], sep[100];
488 long a, b=0, c=0, cn=0;
Marc Kupietza2e64502016-04-27 09:53:51 +0200489 int unmerged;
490
Marc Kupietzdc22b982015-10-09 09:19:34 +0200491 while (1) {
492 st[cn][b] = st1[c];
493 b++;
494 c++;
495 st[cn][b] = 0;
496 if (st1[c] == 0) break;
Marc Kupietz95aa1c02016-03-15 09:40:43 +0100497 if (st1[c] == ' ' || st1[c] == '-') {
498 sep[cn++] = st1[c];
Marc Kupietzdc22b982015-10-09 09:19:34 +0200499 b = 0;
500 c++;
501 }
502 }
503 cn++;
504 for (a = 0; a < cn; a++) {
Marc Kupietza2e64502016-04-27 09:53:51 +0200505 if(search_backw) {
506 for (b = words - 1; b >= 0; b--) if (!strcmp(&vocab[b * max_w], st[a])) break;
507 } else {
508 for (b = 0; b < words; b++) if (!strcmp(&vocab[b * max_w], st[a])) break;
509 }
Marc Kupietz34a3ee92016-02-27 22:43:16 +0100510 if (b == words) b = -1;
Marc Kupietz48c29682016-03-19 11:30:43 +0100511 wl->wordi[a] = b;
512 fprintf(stderr, "Word: \"%s\" Position in vocabulary: %lld\n", st[a], wl->wordi[a]);
Marc Kupietzdc22b982015-10-09 09:19:34 +0200513 if (b == -1) {
Marc Kupietze8da3062016-02-25 08:37:53 +0100514 fprintf(stderr, "Out of dictionary word!\n");
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100515 cn--;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200516 break;
517 }
518 }
Marc Kupietz48c29682016-03-19 11:30:43 +0100519 wl->length=cn;
520 return(wl);
521}
522
523void *_get_neighbours(knnpars *pars) {
524 char *st1 = pars->token;
525 int N = pars->N;
526 long from = pars -> from;
527 unsigned long upto = pars -> upto;
528 char file_name[max_size], st[100][max_size], *sep;
529 float dist, len, *bestd, vec[max_size];
530 long long a, b, c, d, cn, *bi, *besti;
531 char ch;
532 knn *nbs = NULL;
533 wordlist *wl = pars->wl;
534
535 besti = malloc(N * sizeof(long long));
536 bestd = malloc(N * sizeof(float));
537
538 float worstbest=-1;
539
540 for (a = 0; a < N; a++) bestd[a] = 0;
541 a = 0;
542 bi = wl->wordi;
543 cn = wl->length;
544 sep = wl->sep;
545 b = bi[0];
546 c = 0;
Marc Kupietz000ad862016-02-26 14:59:12 +0100547 if (b == -1) {
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100548 N = 0;
549 goto end;
550 }
Marc Kupietzdc22b982015-10-09 09:19:34 +0200551 for (a = 0; a < size; a++) vec[a] = 0;
552 for (b = 0; b < cn; b++) {
553 if (bi[b] == -1) continue;
Marc Kupietz95aa1c02016-03-15 09:40:43 +0100554 if(b>0 && sep[b-1] == '-')
555 for (a = 0; a < size; a++) vec[a] -= M[a + bi[b] * size];
556 else
557 for (a = 0; a < size; a++) vec[a] += M[a + bi[b] * size];
Marc Kupietzdc22b982015-10-09 09:19:34 +0200558 }
559 len = 0;
560 for (a = 0; a < size; a++) len += vec[a] * vec[a];
561 len = sqrt(len);
562 for (a = 0; a < size; a++) vec[a] /= len;
563 for (a = 0; a < N; a++) bestd[a] = -1;
Marc Kupietz000ad862016-02-26 14:59:12 +0100564 for (c = from; c < upto; c++) {
Marc Kupietz5c3887d2016-04-28 08:53:35 +0200565 if(garbage && garbage[c]) continue;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200566 a = 0;
Marc Kupietz34020dc2016-02-25 08:44:19 +0100567// do not skip taget word
Marc Kupietze8da3062016-02-25 08:37:53 +0100568// for (b = 0; b < cn; b++) if (bi[b] == c) a = 1;
569// if (a == 1) continue;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200570 dist = 0;
571 for (a = 0; a < size; a++) dist += vec[a] * M[a + c * size];
Marc Kupietzbe1b9fc2016-02-26 10:34:30 +0100572 if(dist > worstbest) {
573 for (a = 0; a < N; a++) {
574 if (dist > bestd[a]) {
Marc Kupietz33679a32016-03-22 08:49:39 +0100575 memmove(bestd + a + 1, bestd + a, (N - a -1) * sizeof(float));
576 memmove(besti + a + 1, besti + a, (N - a -1) * sizeof(long long));
Marc Kupietzbe1b9fc2016-02-26 10:34:30 +0100577 bestd[a] = dist;
578 besti[a] = c;
579 break;
Marc Kupietzdc22b982015-10-09 09:19:34 +0200580 }
Marc Kupietzdc22b982015-10-09 09:19:34 +0200581 }
Marc Kupietzbe1b9fc2016-02-26 10:34:30 +0100582 worstbest = bestd[N-1];
Marc Kupietzdc22b982015-10-09 09:19:34 +0200583 }
584 }
Marc Kupietz34020dc2016-02-25 08:44:19 +0100585
Marc Kupietz000ad862016-02-26 14:59:12 +0100586 nbs = malloc(sizeof(knn));
587 nbs->index = besti;
588 nbs->dist = bestd;
589 nbs->length = N;
Marc Kupietz44bee3c2016-02-25 16:26:29 +0100590end:
Marc Kupietz000ad862016-02-26 14:59:12 +0100591 pthread_exit(nbs);
Marc Kupietzdc22b982015-10-09 09:19:34 +0200592}
593
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100594
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100595SV *get_neighbours(char *st1, int N, int sort_by, int search_backw, long cutoff) {
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100596 HV *result = newHV();
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100597 float *target_sums, bestd[MAX_NEIGHBOURS], bestn[MAX_NEIGHBOURS], bests[MAX_NEIGHBOURS], vec[max_size];
Marc Kupietza5f60042017-05-04 10:38:12 +0200598 long long old_words;
Marc Kupietz50485ba2016-03-23 09:13:14 +0100599 long besti[MAX_NEIGHBOURS], bestp[MAX_NEIGHBOURS], a, b, c, d, slice;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100600 knn *para_nbs[MAX_THREADS];
601 knn *syn_nbs[MAX_THREADS];
Marc Kupietz000ad862016-02-26 14:59:12 +0100602 knnpars pars[MAX_THREADS];
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100603 pthread_t *pt = (pthread_t *)malloc((num_threads+1) * sizeof(pthread_t));
Marc Kupietz48c29682016-03-19 11:30:43 +0100604 wordlist *wl;
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200605 int syn_threads = (M2? window * 2 : 0);
606 int para_threads = num_threads - syn_threads;
Marc Kupietz48c29682016-03-19 11:30:43 +0100607
Marc Kupietz000ad862016-02-26 14:59:12 +0100608 if(N>MAX_NEIGHBOURS) N=MAX_NEIGHBOURS;
609
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100610 if(cutoff < 1)
611 cutoff=words;
612
Marc Kupietza2e64502016-04-27 09:53:51 +0200613 wl = getTargetWords(st1, search_backw);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100614 if(wl->length < 1)
615 goto end;
Marc Kupietz48c29682016-03-19 11:30:43 +0100616
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100617 old_words = cutoff;
Marc Kupietza5f60042017-05-04 10:38:12 +0200618 if(merge_words > 0)
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100619 cutoff = merge_words * 1.25; /* HACK */
620 slice = cutoff / para_threads;
Marc Kupietza5f60042017-05-04 10:38:12 +0200621
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100622 a = posix_memalign((void **) &target_sums, 128, cutoff * sizeof(float));
623 for(a = 0; a < cutoff; a++)
Marc Kupietzce3d4c62016-03-23 16:11:25 +0100624 target_sums[a] = 0;
625
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200626 printf("Starting %d threads\n", para_threads);
627 fflush(stdout);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100628 for(a=0; a < para_threads; a++) {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100629 pars[a].cutoff = cutoff;
Marc Kupietz000ad862016-02-26 14:59:12 +0100630 pars[a].token = st1;
Marc Kupietz48c29682016-03-19 11:30:43 +0100631 pars[a].wl = wl;
Marc Kupietz000ad862016-02-26 14:59:12 +0100632 pars[a].N = N;
633 pars[a].from = a*slice;
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100634 pars[a].upto = ((a+1)*slice > cutoff? cutoff:(a+1)*slice);
Marc Kupietz000ad862016-02-26 14:59:12 +0100635 pthread_create(&pt[a], NULL, _get_neighbours, (void *) &pars[a]);
636 }
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200637 if(M2) {
638 for(a=0; a < syn_threads; a++) {
Marc Kupietz2c79c5e2017-11-09 16:18:40 +0100639 pars[a + para_threads].cutoff = cutoff;
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200640 pars[a + para_threads].target_sums = target_sums;
641 pars[a + para_threads].wl = wl;
642 pars[a + para_threads].N = N;
643 pars[a + para_threads].from = a;
644 pars[a + para_threads].upto = a+1;
645 pthread_create(&pt[a + para_threads], NULL, getCollocators, (void *) &pars[a + para_threads]);
646 }
Marc Kupietz271e2a42016-03-22 11:37:43 +0100647 }
648 printf("Waiting for para threads to join\n");
649 fflush(stdout);
650 for (a = 0; a < para_threads; a++) pthread_join(pt[a], &para_nbs[a]);
651 printf("Para threads joint\n");
652 fflush(stdout);
Marc Kupietz000ad862016-02-26 14:59:12 +0100653
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200654 /* if(!syn_nbs[0]) */
655 /* goto end; */
Marc Kupietz000ad862016-02-26 14:59:12 +0100656
657 for(b=0; b < N; b++) {
Marc Kupietz271e2a42016-03-22 11:37:43 +0100658 besti[b] = para_nbs[0]->index[b];
659 bestd[b] = para_nbs[0]->dist[b];
Marc Kupietz000ad862016-02-26 14:59:12 +0100660 }
661
Marc Kupietz271e2a42016-03-22 11:37:43 +0100662 for(a=1; a < para_threads; a++) {
663 for(b=0; b < para_nbs[a]->length && para_nbs[a]->index[b] >= 0; b++) {
Marc Kupietza5f60042017-05-04 10:38:12 +0200664 for(c=0; c < N * para_threads; c++) {
Marc Kupietz271e2a42016-03-22 11:37:43 +0100665 if(para_nbs[a]->dist[b] > bestd[c]) {
Marc Kupietz000ad862016-02-26 14:59:12 +0100666 for(d=N-1; d>c; d--) {
667 bestd[d] = bestd[d-1];
668 besti[d] = besti[d-1];
669 }
Marc Kupietz271e2a42016-03-22 11:37:43 +0100670 besti[c] = para_nbs[a]->index[b];
671 bestd[c] = para_nbs[a]->dist[b];
Marc Kupietz000ad862016-02-26 14:59:12 +0100672 break;
673 }
674 }
675 }
676 }
677
Marc Kupietz271e2a42016-03-22 11:37:43 +0100678 AV* array = newAV();
Marc Kupietza5f60042017-05-04 10:38:12 +0200679 int i;
680 int l1_words=0, l2_words=0;
681 for (a = 0, i = 0; i < N && a < 600; a++) {
682 long long c = besti[a];
683 if(merge_words > 0) {
684 if(c >= merge_words) {
685 if(l1_words > N / 2)
686 continue;
687 else
688 l1_words++;
689 } else {
690 if(l2_words > N / 2)
691 continue;
692 else
693 l2_words++;
694 }
695 }
696 fflush(stdout);
697 printf("%s l1:%d l2:%d i:%d a:%ld\n", &vocab[c * max_w], l1_words, l2_words, i, a);
698
Marc Kupietz271e2a42016-03-22 11:37:43 +0100699 HV* hash = newHV();
Marc Kupietza5f60042017-05-04 10:38:12 +0200700 SV* word = newSVpvf(&vocab[c * max_w], 0);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100701 if(latin_enc == 0) SvUTF8_on(word);
Marc Kupietza5f60042017-05-04 10:38:12 +0200702 fflush(stdout);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100703 hv_store(hash, "word", strlen("word"), word , 0);
704 hv_store(hash, "dist", strlen("dist"), newSVnv(bestd[a]), 0);
705 hv_store(hash, "rank", strlen("rank"), newSVuv(besti[a]), 0);
706 AV *vector = newAV();
707 for (b = 0; b < size; b++) {
708 av_push(vector, newSVnv(M[b + besti[a] * size]));
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100709 }
Marc Kupietz271e2a42016-03-22 11:37:43 +0100710 hv_store(hash, "vector", strlen("vector"), newRV_noinc((SV*)vector), 0);
711 av_push(array, newRV_noinc((SV*)hash));
Marc Kupietza5f60042017-05-04 10:38:12 +0200712 i++;
Marc Kupietz271e2a42016-03-22 11:37:43 +0100713 }
714 hv_store(result, "paradigmatic", strlen("paradigmatic"), newRV_noinc((SV*)array), 0);
715
Marc Kupietz50485ba2016-03-23 09:13:14 +0100716 for(b=0; b < MAX_NEIGHBOURS; b++) {
717 besti[b] = -1L;
718 bestd[b] = 0;
719 bestn[b] = 0;
720 bestp[b] = 0;
Marc Kupietz6d9a6782016-03-23 17:25:25 +0100721 bests[b] = 0;
Marc Kupietz50485ba2016-03-23 09:13:14 +0100722 }
723
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200724 if (M2) {
725 printf("Waiting for syn threads to join\n");
726 fflush(stdout);
727 for (a = 0; a < syn_threads; a++) pthread_join(pt[a+para_threads], &syn_nbs[a]);
728 printf("syn threads joint\n");
729 fflush(stdout);
Marc Kupietz50485ba2016-03-23 09:13:14 +0100730
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200731 for(b=0; b < syn_nbs[0]->length; b++) {
732 besti[b] = syn_nbs[0]->index[b];
733 bestd[b] = syn_nbs[0]->dist[b];
734 bestn[b] = syn_nbs[0]->norm[b];
735 bestp[b] = syn_nbs[0]->pos[b];
736 bests[b] = target_sums[syn_nbs[0]->index[b]];
737 }
738
739 if(sort_by != 1) { // sort by responsiveness
740 for(a=1; a < syn_threads; a++) {
741 for(b=0; b < syn_nbs[a]->length; b++) {
742 for(c=0; c < MAX_NEIGHBOURS; c++) {
743 if(syn_nbs[a]->dist[b] > bestd[c]) {
744 for(d=MAX_NEIGHBOURS-1; d>c; d--) {
745 bestd[d] = bestd[d-1];
746 besti[d] = besti[d-1];
747 bestn[d] = bestn[d-1];
748 bestp[d] = bestp[d-1];
749 }
750 besti[c] = syn_nbs[a]->index[b];
751 bestd[c] = syn_nbs[a]->dist[b];
752 bestn[c] = syn_nbs[a]->norm[b];
753 bestp[c] = syn_nbs[a]->pos[b];
754 break;
Marc Kupietz6d9a6782016-03-23 17:25:25 +0100755 }
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200756 }
757 }
758 }
759 } else { // sort by mean p
760 for(a=1; a < syn_threads; a++) {
761 for(b=0; b < syn_nbs[a]->length; b++) {
762 for(c=0; c < MAX_NEIGHBOURS; c++) {
763 if(target_sums[syn_nbs[a]->index[b]] > bests[c]) {
764 for(d=MAX_NEIGHBOURS-1; d>c; d--) {
765 bestd[d] = bestd[d-1];
766 besti[d] = besti[d-1];
767 bestn[d] = bestn[d-1];
768 bestp[d] = bestp[d-1];
769 bests[d] = bests[d-1];
770 }
771 besti[c] = syn_nbs[a]->index[b];
772 bestd[c] = syn_nbs[a]->dist[b];
773 bestn[c] = syn_nbs[a]->norm[b];
774 bestp[c] = syn_nbs[a]->pos[b];
775 bests[c] = target_sums[syn_nbs[a]->index[b]];
776 break;
777 }
Marc Kupietz271e2a42016-03-22 11:37:43 +0100778 }
Marc Kupietz6d9a6782016-03-23 17:25:25 +0100779 }
780 }
781 }
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200782 array = newAV();
783 for (a = 0; a < MAX_NEIGHBOURS && besti[a] >= 0; a++) {
784 HV* hash = newHV();
785 SV* word = newSVpvf(&vocab[besti[a] * max_w], 0);
786 if(latin_enc == 0) SvUTF8_on(word);
787 hv_store(hash, "word", strlen("word"), word , 0);
Marc Kupietzb18978b2017-11-09 14:51:17 +0100788 hv_store(hash, "rank", strlen("rank"), newSVuv(besti[a]), 0);
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200789 hv_store(hash, "dist", strlen("dist"), newSVnv(bestd[a]), 0);
790 hv_store(hash, "norm", strlen("norm"), newSVnv(bestn[a]), 0);
791 hv_store(hash, "sum", strlen("sum"), newSVnv(target_sums[besti[a]]), 0);
792 hv_store(hash, "pos", strlen("pos"), newSVnv(bestp[a]), 0);
793 av_push(array, newRV_noinc((SV*)hash));
Marc Kupietz271e2a42016-03-22 11:37:43 +0100794 }
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200795 hv_store(result, "syntagmatic", strlen("syntagmatic"), newRV_noinc((SV*)array), 0);
Marc Kupietz271e2a42016-03-22 11:37:43 +0100796 }
Marc Kupietz000ad862016-02-26 14:59:12 +0100797end:
Marc Kupietza5f60042017-05-04 10:38:12 +0200798 words = old_words;
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100799 return newRV_noinc((SV*)result);
Marc Kupietz000ad862016-02-26 14:59:12 +0100800}
Marc Kupietz7bc85fd2016-02-24 11:42:41 +0100801
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200802int dump_vecs(char *fname) {
803 long i, j;
804 FILE *f;
805 /* if(words>200000) */
806 /* words=200000; */
Marc Kupietz6b2975c2016-03-18 21:59:33 +0100807
Marc Kupietz43ee87e2016-04-25 10:50:08 +0200808 if((f=fopen(fname, "w")) == NULL) {
809 fprintf(stderr, "cannot open %s for writing\n", fname);
810 return(-1);
811 }
812 fprintf(f, "%lld %lld\n", words, size);
813 for (i=0; i < words; i++) {
814 fprintf(f, "%s ", &vocab[i * max_w]);
815 for(j=0; j < size - 1; j++)
816 fprintf(f, "%f ", M[i*size + j]);
817 fprintf(f, "%f\n", M[i*size + j]);
818 }
819 fclose(f);
820 return(0);
821}
Marc Kupietzdc22b982015-10-09 09:19:34 +0200822