Owl_nlp_lda0
NLP: LDA module
module MS = Owl_sparse.Dok_matrix
module MD = Owl_dense.Matrix.D
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