1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! WordEmbeddings implementations for [`finalfusion::embeddings::Embeddings`].
//! This module is available if the `finalfusion` feature is enabled.
use crate::Float;
use crate::WordEmbeddings;

use finalfusion::embeddings::Embeddings;
use finalfusion::storage::Storage;
use finalfusion::vocab::Vocab;
use ndarray::{CowArray, Ix1};

impl<V, S> WordEmbeddings for Embeddings<V, S>
where
    V: Vocab,
    S: Storage,
{
    fn embedding(&self, word: &str) -> Option<CowArray<Float, Ix1>> {
        self.embedding(word)
    }

    fn embedding_size(&self) -> usize {
        self.dims()
    }
}