diff --git a/src/cc/mallet/topics/HierarchicalLDA.java b/src/cc/mallet/topics/HierarchicalLDA.java index 6fe0d71f9..373815c89 100644 --- a/src/cc/mallet/topics/HierarchicalLDA.java +++ b/src/cc/mallet/topics/HierarchicalLDA.java @@ -459,6 +459,7 @@ public void printState(PrintWriter out) throws IOException { doc++; } + out.close(); } public void printNodes() { diff --git a/src/cc/mallet/topics/tui/HierarchicalLDATUI.java b/src/cc/mallet/topics/tui/HierarchicalLDATUI.java index f5f187384..2e37cc630 100644 --- a/src/cc/mallet/topics/tui/HierarchicalLDATUI.java +++ b/src/cc/mallet/topics/tui/HierarchicalLDATUI.java @@ -111,7 +111,7 @@ public static void main (String[] args) throws java.io.IOException { // Output results if (stateFile.value() != null) { - hlda.printState(new PrintWriter(stateFile.value())); + hlda.printState(new PrintWriter(new BufferedWriter(new FileWriter(stateFile.value())))); } if (testing != null) {