/annoy-rb

annoy-rb provides Ruby bindings for the Annoy (Approximate Nearest Neighbors Oh Yeah).

Primary LanguageC++Apache License 2.0Apache-2.0

annoy-rb

Build Status Gem Version License Documentation

annoy-rb provides Ruby bindings for the Annoy (Approximate Nearest Neighbors Oh Yeah).

Installation

Add this line to your application's Gemfile:

gem 'annoy-rb'

And then execute:

$ bundle install

Or install it yourself as:

$ gem install annoy-rb

Note: annoy-rb does not require the installation of another external library. In addition, annoy-rb does not give any optimization options when building native extensions. If necessary, add optimization options yourself during installation, as follows;

$ bundle config --local build.annoy-rb "--with-cxxflags=-march=native"
$ bundle install

Or:

$ gem install annoy-rb -- --with-cxxflags=-march=native

Documentation

Usage

require 'annoy'

f = 40 # length of item vector that will be indexed.
t = Annoy::AnnoyIndex.new(n_features: f, metric: 'angular')

1000.times do |i|
  v = Array.new(f) { rand }
  t.add_item(i, v)
end

t.build(10) # 10 trees.
t.save('test.ann')

u = Annoy::AnnoyIndex.new(n_features: f, metric: 'angular')
u.load('test.ann')
p u.get_nns_by_item(0, 100) # will find the 100 nearest neighbors.

With the default argument, annoy-rb uses double precision floating point type for the data type of vector element. On the other hand, the Python bindings of Annoy use single precision floating point type. If you want to load a search index created with the Python bindings, specify 'float32' to the dtype argument.

require 'annoy'

f = 40
t = Annoy::AnnoyIndex.new(n_features: f, metric: 'angular', dtype: 'float32')
t.load('index_with_python_bindings.ann')

License

The gem is available as open source under the terms of the Apache-2.0 License.

Contributing

Bug reports and pull requests are welcome on GitHub at https://github.com/yoshoku/annoy-rb. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the Contributor Covenant code of conduct.