module Lda: Owl_nlp_lda
type lda_typ =
| |
SimpleLDA |
| |
FTreeLDA |
| |
LightLDA |
| |
SparseLDA |
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 : float array array; |
|
mutable t_wk : float array array; |
|
mutable t__k : float array; |
|
mutable t__z : int array array; |
|
mutable iter : int; |
|
mutable data : Owl_nlp_corpus.t; |
|
mutable vocb : (string, int) Hashtbl.t; |
}
val include_token : model -> int -> int -> int -> unit
val exclude_token : model -> int -> int -> int -> unit
val show_info : 'a -> 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) Hashtbl.t -> Owl_nlp_corpus.t -> model
val train : lda_typ -> model -> unit