Improve hack for collocates list end
Change-Id: Ia869fdeb7856a0c330620d3bb36987380484cb67
diff --git a/Project.toml b/Project.toml
index db84e29..cf2f458 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
name = "DerekoVecs"
uuid = "d522d5f0-9ae6-4a1c-b144-42ce9f15cae4"
authors = ["Marc Kupietz"]
-version = "0.5.0"
+version = "0.5.1"
[deps]
ArtifactUtils = "8b73e784-e7d8-4ea5-973d-377fed4e3bce"
diff --git a/src/DerekoVecs.jl b/src/DerekoVecs.jl
index fa22a53..cef4cec 100644
--- a/src/DerekoVecs.jl
+++ b/src/DerekoVecs.jl
@@ -199,7 +199,7 @@
i = 0
for c in unsafe_wrap(Vector{collocate}, res, 1000, own=false)
- if (c.w2 <= 0 || c.w2 > max_vocab_index)
+ if (c.w2 <= 0 || c.w2 > max_vocab_index || c.f2 <= 1 || c.pmi < 0.01)
break
end
i += 1