tangram_text 0.7.0

Tangram makes it easy for programmers to train, deploy, and monitor machine learning models.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use fnv::FnvHashMap;

#[derive(Clone, Debug)]
pub struct WordEmbeddingModel {
	pub size: usize,
	pub words: FnvHashMap<String, usize>,
	pub values: Vec<f32>,
}

impl WordEmbeddingModel {
	/// Get the word embedding for `word` if it is present in the model.
	pub fn get(&self, word: &str) -> Option<&[f32]> {
		let index = self.words.get(word)?;
		let embedding = self
			.values
			.get(index * self.size..index * self.size + self.size)?;
		Some(embedding)
	}
}