from dt import *
from java.util import *;
dt=DecisionTree()

dt.setAttributes(["Outlook", "Temperature","Humidity", "Wind"])
dt.addExample(["Sunny",   "Hot",  "High",   "Weak"],0)
dt.addExample(["Sunny",    "Hot",  "High",   "Strong"],0)
dt.addExample(["Overcast", "Hot",  "High",   "Weak"],1)
dt.addExample(["Rain",     "Mild", "High",   "Weak"],1)
dt.addExample(["Rain",     "Cool", "Normal", "Weak"],1)
dt.addExample(["Rain",     "Cool", "Normal", "Strong"],0)
dt.addExample(["Overcast", "Cool", "Normal", "Strong"],1)
dt.addExample(["Sunny",    "Mild", "High",   "Weak"],0)
dt.addExample(["Sunny",    "Cool", "Normal", "Weak"],1)
dt.addExample(["Rain",     "Mild", "Normal", "Weak"],1)
dt.addExample(["Sunny",    "Mild", "Normal", "Strong"],1)
dt.addExample(["Overcast", "Mild", "High",   "Strong"],1)
dt.addExample(["Overcast", "Hot",  "Normal", "Weak"],1)
dt.addExample(["Rain",     "Mild", "High",   "Strong"],0);


case1=HashMap()
case1.put("Outlook", "Overcast")
case1.put("Temperature", "Hot")
case1.put("Humidity", "High")
case1.put("Wind", "Strong")
print dt.apply(case1)


case1=HashMap()
case1.put("Outlook", "Sunny")
case1.put("Temperature", "Mild")
case1.put("Humidity", "High")
case1.put("Wind", "Weak")
print dt.apply(case1)