1"""
2Created on May 26, 2015
3
4@author: jhkwakkel
5"""
6
7import matplotlib.pyplot as plt
8
9import ema_workbench.analysis.cart as cart
10from ema_workbench import ema_logging, load_results
11
12ema_logging.log_to_stderr(level=ema_logging.INFO)
13
14default_flow = 2.178849944502783e7
15
16# load data
17fn = "./data/5000 runs WCM.tar.gz"
18results = load_results(fn)
19x, outcomes = results
20
21ooi = "throughput Rotterdam"
22outcome = outcomes[ooi] / default_flow
23y = outcome < 1
24
25cart_alg = cart.CART(x, y)
26cart_alg.build_tree()
27
28# print cart to std_out
29print(cart_alg.stats_to_dataframe())
30print(cart_alg.boxes_to_dataframe())
31
32# visualize
33cart_alg.show_boxes(together=False)
34cart_alg.show_tree()
35plt.show()