sif_embedding/
finalfusion.rs

1//! WordEmbeddings implementations for [`finalfusion::embeddings::Embeddings`].
2//! This module is available if the `finalfusion` feature is enabled.
3use crate::Float;
4use crate::WordEmbeddings;
5
6use finalfusion::embeddings::Embeddings;
7use finalfusion::storage::Storage;
8use finalfusion::vocab::Vocab;
9use ndarray::{CowArray, Ix1};
10
11impl<V, S> WordEmbeddings for Embeddings<V, S>
12where
13    V: Vocab,
14    S: Storage,
15{
16    fn embedding(&self, word: &str) -> Option<CowArray<Float, Ix1>> {
17        self.embedding(word)
18    }
19
20    fn embedding_size(&self) -> usize {
21        self.dims()
22    }
23}