[−][src]Trait finalfusion::similarity::AnalogyBy
Trait for analogy queries with a custom similarity function.
Required methods
fn analogy_by_masked<F>(
&self,
query: [&str; 3],
remove: [bool; 3],
limit: usize,
similarity: F
) -> Result<Vec<WordSimilarityResult>, [bool; 3]> where
F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>,
&self,
query: [&str; 3],
remove: [bool; 3],
limit: usize,
similarity: F
) -> Result<Vec<WordSimilarityResult>, [bool; 3]> where
F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>,
Perform an analogy query using the given similarity function.
This method returns words that are close in vector space the analogy
query word1
is to word2
as word3
is to ?
. More concretely,
it searches embeddings that are similar to:
embedding(word2) - embedding(word1) + embedding(word3)
At most, limit
results are returned.
remove
specifies which parts of the queries are excluded from the
output candidates. If remove[0]
is true
, word1
cannot be
returned as an answer to the query.
Result::Err
is returned when no embedding could be computed
for one or more of the tokens, indicating which of the tokens
were present.
Provided methods
fn analogy_by<F>(
&self,
query: [&str; 3],
limit: usize,
similarity: F
) -> Result<Vec<WordSimilarityResult>, [bool; 3]> where
F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>,
&self,
query: [&str; 3],
limit: usize,
similarity: F
) -> Result<Vec<WordSimilarityResult>, [bool; 3]> where
F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>,
Perform an analogy query using the given similarity function.
This method returns words that are close in vector space the analogy
query word1
is to word2
as word3
is to ?
. More concretely,
it searches embeddings that are similar to:
embedding(word2) - embedding(word1) + embedding(word3)
At most, limit
results are returned.
Result::Err
is returned when no embedding could be computed
for one or more of the tokens, indicating which of the tokens
were present.
Implementors
impl<V, S> AnalogyBy for Embeddings<V, S> where
V: Vocab,
S: StorageView,
[src]
V: Vocab,
S: StorageView,