projector (version 0.0.2)

get_annoy_model: Build Annoy model

Description

RcppAnnoy model is used to retrieve the most similar vectors to a pivot one. This function builds the RcppAnnoy model.

Usage

get_annoy_model(vectors, number_trees)

Arguments

vectors

matrix where each row is an observation. rownames should contain textual versions of the vectors.

number_trees

integer counting the number of trees to grow in Annoy (for neighbor search). More gives better results but is slower to compute.

Examples

Run this code
# NOT RUN {
if (interactive()){
# This example should be run with a higher quality model
# than the one embedded in fastrtext
library(projector)
library(fastrtext)

model_test_path <- system.file("extdata",
                               "model_unsupervised_test.bin",
                               package = "fastrtext")
model <- load_model(model_test_path)
word_embeddings <- get_word_vectors(model,
                                    words = head(get_dictionary(model), 2e5))

annoy_model <- get_annoy_model(word_embeddings, 5)
}
# }

Run the code above in your browser using DataCamp Workspace