From d66c994b216f1c6fbab5192937280d7ab5a8342c Mon Sep 17 00:00:00 2001 From: John Myles White Date: Fri, 6 Dec 2013 18:55:57 -0800 Subject: [PATCH] Minimal changes to make example work --- examples/LDA.jl | 16 +++++++++++----- src/TopicModels.jl | 7 ++++++- 2 files changed, 17 insertions(+), 6 deletions(-) diff --git a/examples/LDA.jl b/examples/LDA.jl index d952bae..16f7183 100644 --- a/examples/LDA.jl +++ b/examples/LDA.jl @@ -1,8 +1,14 @@ -load("TopicModels") using TopicModels -testDocuments = readDocuments(open("cora.documents")) -testLexicon = readLexicon(open("cora.lexicon")) -model = Model(fill(0.1, 10), 0.01, length(testLexicon), testDocuments) -trainModel(testDocuments, model, 30) +exdir = Pkg.dir("TopicModels", "examples") + +testDocuments = readDocuments(open(joinpath(exdir, "cora.documents"))) +testLexicon = readLexicon(open(joinpath(exdir, "cora.lexicon"))) + +corpus = Corpus(testDocuments) + +model = Model(fill(0.1, 10), 0.01, length(testLexicon), corpus) + +@time trainModel(model, 30) + topWords = topTopicWords(model, testLexicon, 21) diff --git a/src/TopicModels.jl b/src/TopicModels.jl index 6cc6679..17ccde8 100644 --- a/src/TopicModels.jl +++ b/src/TopicModels.jl @@ -193,6 +193,11 @@ function readLexicon(stream) map(chomp, convert(Array{String,1}, lines)) end -export trainModel, topTopicWords, readDocuments, readLexicon +export Corpus, + Model, + readDocuments, + readLexicon, + topTopicWords, + trainModel end