diff options
| author | metamuffin <metamuffin@disroot.org> | 2023-09-20 22:15:48 +0200 |
|---|---|---|
| committer | metamuffin <metamuffin@disroot.org> | 2023-09-20 22:15:48 +0200 |
| commit | 4c7c58f487d0ccb70162421e5a871a4020454022 (patch) | |
| tree | c50397cf990c99e61e6a45a7703737faea32f6ca /src/main.rs | |
| parent | d0ce1d9134968a15d37135622138f6b8b7667454 (diff) | |
| download | embeddings-sort-4c7c58f487d0ccb70162421e5a871a4020454022.tar embeddings-sort-4c7c58f487d0ccb70162421e5a871a4020454022.tar.bz2 embeddings-sort-4c7c58f487d0ccb70162421e5a871a4020454022.tar.zst | |
use angular distance instead of cossim replacement
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) |