From 4c7c58f487d0ccb70162421e5a871a4020454022 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Wed, 20 Sep 2023 22:15:48 +0200 Subject: use angular distance instead of cossim replacement --- src/main.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index 2dda2da..f9bf4fc 100644 --- a/src/main.rs +++ b/src/main.rs @@ -20,7 +20,7 @@ enum Embedder { Hue, Color, ContentEuclidean, - ContentCosineSim, + ContentAngularDistance, ContentManhatten, } @@ -177,8 +177,8 @@ fn main() -> Result<()> { Embedder::Brightness => process_embedder(BrightnessEmbedder, &args, &cfg), Embedder::Hue => process_embedder(HueEmbedder, &args, &cfg), Embedder::Color => process_embedder(ColorEmbedder, &args, &cfg), - Embedder::ContentCosineSim => { - process_embedder(ContentEmbedder::::new(&cfg), &args, &cfg) + Embedder::ContentAngularDistance => { + process_embedder(ContentEmbedder::::new(&cfg), &args, &cfg) } Embedder::ContentEuclidean => { process_embedder(ContentEmbedder::::new(&cfg), &args, &cfg) -- cgit v1.2.3-70-g09d2