nlp/com.londogard.nlp.meachinelearning.transformers/Bm25Transformer

Bm25Transformer

public final class Bm25Transformer implements Transformer<Float, Float>

Constructors

Bm25Transformer

public Bm25Transformer Bm25Transformer(Integer k, Float b)

Functions

public Unit fit(MultiArray<Float, D2> input)

public MultiArray<Float, D2> fitTransform(MultiArray<Float, D2> input)

public final Float getB()

public final NDArray<Float, D1> getIdf()

public final Integer getK()

public final Unit setIdf(NDArray<Float, D1> idf)

public D2SparseArray transform(MultiArray<Float, D2> input)

Input is a count matrix

Properties

private final Float b

private NDArray<Float, D1> idf

private final Integer k