Module emb

cosine(A::binary(), B::binary()) -> float()

dim(X1::encoder()) -> pos_integer()

dot(A::binary(), B::binary()) -> float()

encode(X1::encoder(), Text::binary()) -> {ok, binary()} | {error, term()}

encode_batch(E::encoder(), Texts::[binary()]) -> {ok, [binary()]} | {error, term()}

index(Ix::kvex:index(), E::encoder(), Id::kvex:id(), Text::binary()) -> ok | {error, term()}

index_batch(Ix::kvex:index(), E::encoder(), Pairs::[{kvex:id(), binary()}]) -> ok | {error, term()}

load(Opts::#{tokenizer := file:filename(), model := file:filename(), pooling => mean | cls | none, normalize => boolean(), output_name => binary()}) -> {ok, encoder()} | {error, term()}

new_index(X1::encoder()) -> {ok, kvex:index()}

search(Ix::kvex:index(), E::encoder(), Query::binary(), K::pos_integer()) -> {ok, [{kvex:id(), float()}]} | {error, term()}

unload(X1::encoder()) -> ok