beet_ml 0.0.8

Machine Learning actions built upon beet_flow
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use beet_core::prelude::*;
use beet_ml::prelude::*;

fn main() -> Result {
	let mut bert = async_io::task::block_on(Bert::new(BertConfig::default()))?;
	let embeddings = bert.get_embeddings(vec![
		"The cat sits outside".into(),
		"A man is playing guitar".into(),
		"I love pasta".into(),
		"The new movie is awesome".into(),
		"The cat plays in the garden".into(),
		"A woman watches TV".into(),
		"The new movie is so great".into(),
		"Do you like pizza?".into(),
	])?;

	let target_index = 0;

	let results = embeddings.scores_sorted(target_index)?;
	assert_eq!(
		embeddings.sentences[results[0].0].as_ref(),
		"The cat plays in the garden"
	);

	println!("Results: {:?}", results);

	Ok(())
}