Untitled notebook
Mix.install([
{:ex_final_fusion, "~> 0.1.0"}
])
Section
{:ok, ref} = ExFinalFusion.read("~/cc.pl.300.bin", :fasttext)
{:ok, #ExFinalFusion<0.994911395.4139909124.42479>}
query = ["pies", "kot", "lis"]
{:ok, {vec, _}} = ExFinalFusion.mean_embedding_batch(ref, query)
ExFinalFusion.embedding_similarity(ref, vec, skip: query, limit: 5)
{:ok,
[
{"piesek", 0.7294180393218994},
{"Pies", 0.7123726606369019},
{"kocur", 0.7055572271347046},
{"kotek", 0.6911299824714661},
{"zwierzak", 0.6870898008346558}
]}
ExFinalFusion.analogy(ref, "papuga", "pies", "młotek", limit: 5)
{:ok,
[
{"nóż", 0.4930921494960785},
{"Pies", 0.4844633936882019},
{"kij", 0.48363009095191956},
{"piesek", 0.4727526009082794},
{"młotekmaj", 0.457378089427948}
]}
ExFinalFusion.word_similarity(ref, "kaczka", limit: 5)
{:ok,
[
{"kaczki", 0.646961510181427},
{"Kaczka", 0.6459450721740723},
{"kaczką", 0.6333967447280884},
{"gęś", 0.6245576739311218},
{"kaczkę", 0.6081993579864502}
]}