Decision Tree Visualization

[weixuanfu]

Visualization of decision tree.

[weixuanfu]

I generated a demo of visualization of decision tree. Should we use this for top 10 important features?

[hjwilli]

This looks great! Yes, let's try with the top 10.

[weixuanfu]

• max depth as a environment variable: add environment variable named DT_MAX_DEPTH
• holdout score from the DS: add dtree_test_score into value.json for this score
• difference in holdout scores: add abs_diff_test_score into value.json for absolute value of the difference

[weixuanfu]

@hjwilli in the result folder, two files named dtree_{ID}.dot and dtree_{ID}.png respectively are for decision tree visualization. The .dot may be used for web UI (check this link)