probability compute boilerplate

David5L · Nov 19, 2013 · 22dc446 · 22dc446
1 parent 04d50a9
commit 22dc446
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/main/java/hex/gbm/DTree.java b/src/main/java/hex/gbm/DTree.java
@@ -558,6 +558,11 @@ public void generateHTML(String title, StringBuilder sb) {
         "class=\'btn btn-inverse btn-mini\'>Java Model</a></div><br /><div class=\"hide\" id=\"javaModel\">"       +
         "<pre style=\"overflow-y:scroll;\"><code class=\"language-java\">"+DocGen.HTML.escape2(toJava())+"</code></pre></div>");
       }
+      File file = new File("/home/spencer/GBMScore.java");
+      FileWriter fw = new FileWriter(file.getAbsoluteFile());
+      BufferedWriter bw = new BufferedWriter(fw);
+      bw.write(toJava());
+      bw.close();
       String[] domain = _domains[_domains.length-1]; // Domain of response col
 
       // Top row of CM
@@ -868,6 +873,10 @@ StringBuilder toString(final String res, CompressedTree ct, final StringBuilder
           }.visit();
           sb.p(";\n");
         }
+        sb.indent(2).p("// Compute Probabilities\n");
+        sb.indent(2).p("float sum = 0;\n");
+        sb.indent(2).p("for(int i=1;i<preds.length; i++) sum += Math.exp(preds[i]);\n");
+        sb.indent(2).p("for(int i=1; i<preds.length; i++) preds[i] = (float)Math.exp(preds[i]) / sum;\n");
       }
     }
   }