import monkdata as m
import dtree as d



monk = [m.monk1, m.monk2, m.monk3]
monktest = [m.monk1test, m.monk2test, m.monk3test]

#Assignment 1
print("Entropy training data")
for x in range(len(monktest)) : print("MONK", x+1, " :", d.entropy(monktest[x]))

#Assignment 3
for x in range(len(monk)):
    print("\ninformationgain monk", x+1)
    for a in range(len(m.attributes)):
        print("Attribute", m.attributes[a].name, " :", d.averageGain(monk[x], m.attributes[a]))
print("\nBest attributes")
for x in range(len(monk)):
    print("MONK", x+1, ":", d.bestAttribute(monk[x], m.attributes))



#Assignment 5
m1 = d.select(m.monk1, m.attributes[4], True)
m1_gain = d.averageGain(m1, m.attributes[4])
print(m1_gain)
print(d.entropy(m1))