Module Owl_nlp_lda0

NLP: LDA module

module MD = Owl_dense.Matrix.D
type lda_typ =
| SimpleLDA
| FTreeLDA
| LightLDA
| SparseLDA
type dsmat = ( float, Stdlib.Bigarray.float64_elt ) Owl_dense_matrix_generic.t
type spmat = ( float, Stdlib.Bigarray.float64_elt ) Owl_sparse_dok_matrix.t
type model = {
mutable n_d : int;
mutable n_k : int;
mutable n_v : int;
mutable alpha : float;
mutable beta : float;
mutable alpha_k : float;
mutable beta_v : float;
mutable t_dk : dsmat;
mutable t_wk : spmat;
mutable t__k : dsmat;
mutable t__z : int array array;
mutable iter : int;
mutable data : int array array;
mutable vocb : ( string, int ) Stdlib.Hashtbl.t;
}
val include_token : model -> int -> int -> int -> unit
val exclude_token : model -> int -> int -> int -> unit
val likelihood : model -> float
val show_info : model -> int -> float -> unit
module SimpleLDA : sig ... end
module SparseLDA : sig ... end
module FTreeLDA : sig ... end
module LightLDA : sig ... end
val init : ?iter:int -> int -> ( string, int ) Stdlib.Hashtbl.t -> int array array -> model
val train : lda_typ -> model -> unit