Esempio n. 1
0
  /** Print the model. */
  public void printModel(String outputDirectory) {
    try {
      String domain = model.param.domain;
      printModelParameters(model.param, outputDirectory + domain + modelParamSuffix);
      printTopicWordAssignment(model.z, model.corpus, outputDirectory + domain + tassignSuffix);
      printDocumentTopicDistribution(
          model.getDocumentTopicDistrbution(), outputDirectory + domain + documentTopicDistSuff);
      printTopicWordDistribution(
          model.getTopicWordDistribution(), outputDirectory + domain + topicWordDistSuff);
      ArrayList<ArrayList<ItemWithValue>> topWordsUnderTopics =
          model.getTopWordStrsWithProbabilitiesUnderTopics(model.param.twords);
      printTopWordsUnderTopics(topWordsUnderTopics, outputDirectory + domain + twordsSuffix);
      printDocs(model.corpus.docs, outputDirectory + domain + docsSuffix);
      printVocabulary(model.corpus.vocab, outputDirectory + domain + vocabSuffix);

      // For knowledge-based topic models only.
      model.printKnowledge(outputDirectory + domain + knowledgeSuffix);
    } catch (Exception ex) {
      System.out.println("Error while printing the topic model: " + ex.getMessage());
      ex.printStackTrace();
    }
  }