diff --git a/lib/reckon/cosine_similarity.rb b/lib/reckon/cosine_similarity.rb index dc6a198..7c2300a 100644 --- a/lib/reckon/cosine_similarity.rb +++ b/lib/reckon/cosine_similarity.rb @@ -1,4 +1,5 @@ require 'matrix' +require 'set' # Implementation of consine similarity using TF-IDF for vectorization. # Used to suggest which account a transaction should be assigned to