Module Owl_nlp_lda

NLP: LDA module

Type definition
type lda_typ =
  1. | SimpleLDA
  2. | FTreeLDA
  3. | LightLDA
  4. | SparseLDA
    (*

    Type of LDA training algorithms.

    *)
type model

Type of LDA model.

Core functions
val init : ?iter:int -> int -> (string, int) Hashtbl.t -> Owl_nlp_corpus.t -> model

``init ~iter k v d`` inits an LDA model for training. The default iteration is 100.

Parameters: * ``iter``: number of iterations. * ``k``: number of topics. * ``v``: vocabulary. * ``d``: corpus.

val train : lda_typ -> model -> unit

After calling ``init``, calling this function starts the training.

Helper functions
val show_info : model -> int -> float -> unit

Function for printing out log information, tailored for LDA training.

val include_token : model -> int -> int -> int -> unit

Include a token in model, used in training and you are not supposed to use it.

val exclude_token : model -> int -> int -> int -> unit

Exclude a token in model, used in training and you are not supposed to use it.