Skip to content

W
word2vec
Commit dd2d92a3 authored by tmikolov's avatar tmikolov
Browse files
Options

fixed bug: the counter in K-means was updated too frequently

parent 445fc1e4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
word2vec.c
View file @ dd2d92a3
...@@ -561,11 +561,9 @@ void TrainModel() { ...@@ -561,11 +561,9 @@ void TrainModel() {
for (b = 0; b < clcn * layer1_size; b++) cent[b] = 0; for (b = 0; b < clcn * layer1_size; b++) cent[b] = 0;
for (b = 0; b < clcn; b++) centcn[b] = 1; for (b = 0; b < clcn; b++) centcn[b] = 1;
for (c = 0; c < vocab_size; c++) { for (c = 0; c < vocab_size; c++) {
for (d = 0; d < layer1_size; d++) { for (d = 0; d < layer1_size; d++) cent[layer1_size * cl[c] + d] += syn0[c * layer1_size + d];
cent[layer1_size * cl[c] + d] += syn0[c * layer1_size + d];
centcn[cl[c]]++; centcn[cl[c]]++;
} }
}
for (b = 0; b < clcn; b++) { for (b = 0; b < clcn; b++) {
closev = 0; closev = 0;
for (c = 0; c < layer1_size; c++) { for (c = 0; c < layer1_size; c++) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment