bm25

Types

BM25Index

pub type BM25Index {
  BM25Index(
    term_freq: dict.Dict(String, dict.Dict(fact.EntityId, Int)),
    doc_freq: dict.Dict(String, Int),
    doc_len: dict.Dict(fact.EntityId, Int),
    avg_doc_len: Float,
    doc_count: Int,
    attribute: String,
  )
}

Constructors

BM25Index(
  term_freq: dict.Dict(String, dict.Dict(fact.EntityId, Int)),
  doc_freq: dict.Dict(String, Int),
  doc_len: dict.Dict(fact.EntityId, Int),
  avg_doc_len: Float,
  doc_count: Int,
  attribute: String,
)

Values

add

pub fn add(
  index: BM25Index,
  entity: fact.EntityId,
  text: String,
) -> BM25Index

build

pub fn build(
  datoms: List(fact.Datom),
  attribute: String,
) -> BM25Index

empty

pub fn empty(attribute: String) -> BM25Index

remove

pub fn remove(
  index: BM25Index,
  entity: fact.EntityId,
  text: String,
) -> BM25Index

score

pub fn score(
  index: BM25Index,
  entity: fact.EntityId,
  query: String,
  k1: Float,
  b: Float,
) -> Float