diff options
Diffstat (limited to 'src/main.rs')
-rw-r--r-- | src/main.rs | 6 |
1 files changed, 3 insertions, 3 deletions
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::<CosineSimilarity>::new(&cfg), &args, &cfg) + Embedder::ContentAngularDistance => { + process_embedder(ContentEmbedder::<AngularDistance>::new(&cfg), &args, &cfg) } Embedder::ContentEuclidean => { process_embedder(ContentEmbedder::<EuclidianDistance>::new(&cfg), &args, &cfg) |