List of usage examples for org.apache.mahout.clustering.lda.cvb InMemoryCollapsedVariationalBayes0 main
public static void main(String[] args) throws Exception
From source file:org.plista.kornakapi.core.training.LDATopicFactorizer.java
License:Apache License
/** * //from w ww. java2 s .com * @return * @throws TasteException */ public SemanticModel factorize() throws TasteException, IOException { List<String> argList = Lists.newLinkedList(); argList.add("-i"); argList.add(sparseVectorIn.toString() + "/matrix"); argList.add("-to"); argList.add(topicsOut.toString()); argList.add("--numTopics"); argList.add(k.toString()); argList.add("-d"); argList.add(((LDARecommenderConfig) conf).getTopicsDictionaryPath()); argList.add("--alpha"); argList.add(alpha.toString()); argList.add("--eta"); argList.add(eta.toString()); argList.add("-do"); argList.add(((LDARecommenderConfig) conf).getLDADocTopicsPath()); argList.add("-c"); argList.add(convergenceDelta.toString()); argList.add("-ntt"); argList.add(((LDARecommenderConfig) conf).getTrainingThreats().toString()); argList.add("-m"); argList.add(((LDARecommenderConfig) conf).getMaxIterations()); argList.add("-nut"); argList.add(((LDARecommenderConfig) conf).getTrainingThreats().toString()); String[] args = argList.toArray(new String[argList.size()]); try { InMemoryCollapsedVariationalBayes0.main(args); //computeAllTopicPosterior(); getAllTopicPosterior(); } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } return new SemanticModel(indexItem, itemIndex, itemFeatures, new Path(((LDARecommenderConfig) conf).getLDARecommenderModelPath()), conf); }