Module Owl_nlp_lda0Source

NLP: LDA module

Sourcetype lda_typ =
  1. | SimpleLDA
  2. | FTreeLDA
  3. | LightLDA
  4. | SparseLDA
Sourcetype model = {
  1. mutable n_d : int;
  2. mutable n_k : int;
  3. mutable n_v : int;
  4. mutable alpha : float;
  5. mutable beta : float;
  6. mutable alpha_k : float;
  7. mutable beta_v : float;
  8. mutable t_dk : dsmat;
  9. mutable t_wk : spmat;
  10. mutable t__k : dsmat;
  11. mutable t__z : int array array;
  12. mutable iter : int;
  13. mutable data : int array array;
  14. mutable vocb : (string, int) Hashtbl.t;
}
Sourceval include_token : model -> int -> int -> int -> unit
Sourceval exclude_token : model -> int -> int -> int -> unit
Sourceval likelihood : model -> float
Sourceval show_info : model -> int -> float -> unit
Sourcemodule SimpleLDA : sig ... end
Sourcemodule SparseLDA : sig ... end
Sourcemodule FTreeLDA : sig ... end
Sourcemodule LightLDA : sig ... end
Sourceval init : ?iter:int -> int -> (string, int) Hashtbl.t -> int array array -> model
Sourceval train : lda_typ -> model -> unit