commit 6327bd0d80a902c502d0851a607238744f36613f
author Marc Kupietz <kupietz@ids-mannheim.de> Sun Jul 31 10:13:46 2022 +0200
committer Marc Kupietz <kupietz@ids-mannheim.de> Sun Jul 31 10:26:29 2022 +0200
tree 7b1cff68486bb99e89f56e076fcea6d4a0300031
parent 370dcb5750c42fa2191cf9d2ce8dbc6b2051c370

Fix stop criterion for getting collocates

Change-Id: I3e934a805f7ac91a47f7122c8b802d1397b2aec0

src/DerekoVecs.jl

diff --git a/src/DerekoVecs.jl b/src/DerekoVecs.jl
index ae728c2..e71d01e 100644
--- a/src/DerekoVecs.jl
+++ b/src/DerekoVecs.jl
@@ -198,9 +198,11 @@
     end
 
     i = 0
-    for c in unsafe_wrap(Vector{collocate}, res, max, own=false)
+    unsafe_array = unsafe_wrap(Vector{collocate}, res, max, own=false)
+    window = unsafe_array[1].window
+    for c in unsafe_array
         i += 1
-        if (c.w2 <= 0 || c.w2 > max_vocab_index || c.f2 <= 1 || c.pmi < 0.01)
+        if (c.w2 < 0 || c.w2 > max_vocab_index || c.window != window)
             i -= 1
             break
         end