def saveDictionary(trend, dictionary):
dictName = globalobjs.getLdaObjFileName(trend, globalobjs.DICT)
if dictName == "":
return False
else:
dictionary.save(dictName)
return True
def saveBOWCorpus(trend, bowcorpus, extn = globalobjs.MM_CORPUS_TYPE):
corpusName = globalobjs.getLdaObjFileName(trend, globalobjs.CORPUS, extn = extn)
if corpusName == "":
return False
if(extn == globalobjs.MM_CORPUS_TYPE):
corpora.MmCorpus.serialize(corpusName, bowcorpus)
elif(extn == globalobjs.BLEI_CORPUS_TYPE):
corpora.BleiCorpus.serialize(corpusName, bowcorpus)
elif(extn == globalobjs.SVMLIGHT_CORPUS_TYPE):
corpora.SvmLightCorpus.serialize(corpusName, bowcorpus)
elif(extn == globalobjs.LOW_CORPUS_TYPE):
corpora.LowCorpus.serialize(corpusName, bowcorpus)
def saveLdaModel(trend, ldamodel, **kwargs):
updates, topics, optional, opt_suffix = "", "", "", ""
if "updates" in kwargs:
updates = kwargs['updates']
opt_suffix += "-up" + str(updates)
if "topics" in kwargs:
topics = kwargs['topics']
opt_suffix += "-topics" + str(topics)
if "optional" in kwargs:
optional = kwargs['optional']
opt_suffix += str(optional)
optStr = "_online_"
if lda.numupdates == 0:
optStr = "_batch_"
modelName = globalobjs.getLdaObjFileName(trend + opt_suffix + optStr, globalobjs.MODEL)
if modelName == "":
return False
ldamodel.save(modelName)
