From 9ba0e12332115857d296a9b3042a03d9bf6f303a Mon Sep 17 00:00:00 2001
From: Arno Candel <arno.candel@gmail.com>
Date: Tue, 17 Mar 2015 14:37:33 -0700
Subject: [PATCH] Write trees to individual files for easier debugging.

---
 .../java/hex/gbm/SharedTreeModelBuilder.java   | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/main/java/hex/gbm/SharedTreeModelBuilder.java b/src/main/java/hex/gbm/SharedTreeModelBuilder.java
index 85d39a13ad..0b45ad9285 100644
--- a/src/main/java/hex/gbm/SharedTreeModelBuilder.java
+++ b/src/main/java/hex/gbm/SharedTreeModelBuilder.java
@@ -23,6 +23,9 @@
 import water.util.ModelUtils;
 import water.util.Utils;
 
+import java.io.FileNotFoundException;
+import java.io.PrintWriter;
+import java.io.UnsupportedEncodingException;
 import java.util.Arrays;
 import java.util.Random;
 
@@ -938,11 +941,22 @@ public static Random createRNG(long seed) {
     return new MersenneTwisterRNG(new int[] { (int)(seed>>32L),(int)seed });
   }
 
+  static int counter = 0;
   // helper for debugging
   static protected void printGenerateTrees(DTree[] trees) {
     for( int k=0; k<trees.length; k++ )
-      if( trees[k] != null )
-        System.out.println(trees[k].root().toString2(new StringBuilder(),0));
+      if( trees[k] != null ) {
+        try {
+          PrintWriter writer = new PrintWriter("/tmp/h2o.tree" + ++counter + ".txt", "UTF-8");
+          writer.println(trees[k].root().toString2(new StringBuilder(), 0));
+          writer.close();
+        } catch (FileNotFoundException e) {
+          e.printStackTrace();
+        } catch (UnsupportedEncodingException e) {
+          e.printStackTrace();
+        }
+        System.out.println(trees[k].root().toString2(new StringBuilder(), 0));
+      }
   }
 
   protected final void debugPrintTreeColumns(Frame fr) {