diff --git a/src/main/java/de/uni_leipzig/informatik/asv/hdp/HDPGibbsSampler.java b/src/main/java/de/uni_leipzig/informatik/asv/hdp/HDPGibbsSampler.java index 87bb696..b25c3e0 100644 --- a/src/main/java/de/uni_leipzig/informatik/asv/hdp/HDPGibbsSampler.java +++ b/src/main/java/de/uni_leipzig/informatik/asv/hdp/HDPGibbsSampler.java @@ -204,7 +204,7 @@ protected void removeWord(int docID, int i){ if (docState.wordCountByTable[table] == 0) { // table is removed totalNumberOfTables--; numberOfTablesByTopic[k]--; - docState.tableToTopic[table] --; + docState.tableToTopic[table] = -1; } } @@ -414,4 +414,4 @@ public static void main(String[] args) throws IOException { } -} \ No newline at end of file +}