def testDummySet2(setFunc=setEntropy, infoFunc=infoGain): """Correct classification rate is 0.55""" examples, attrValues, labelName, labelValues = getDummyDataset2() print 'Testing dummy dataset 2. Number of examples %d.' % len(examples) tree = makeTree(examples, attrValues, labelName, setFunc, infoFunc) print 'Tree is as follows:\n%s\n' % str(tree) print 'Tree size: %d.\n' % tree.count() examples, attrValues, labelName, labelValues = getDummyDataset2(test=True) evaluation = evaluateTree(tree, examples, labelName) print 'Results for training set:\n%s\n' % str(evaluation) printDemarcation() return (tree, evaluation)
def testDummySet2(setFunc = setEntropy, infoFunc = infoGain): """Correct classification rate is 0.55""" examples,attrValues,labelName,labelValues = getDummyDataset2() print 'Testing dummy dataset 2. Number of examples %d.'%len(examples) tree = makeTree(examples, attrValues, labelName, setFunc, infoFunc) print 'Tree is as follows:\n%s\n'%str(tree) print 'Tree size: %d.\n'%tree.count() examples,attrValues,labelName,labelValues = getDummyDataset2(test=True) evaluation = evaluateTree(tree,examples,labelName) print 'Results for training set:\n%s\n'%str(evaluation) printDemarcation() return (tree,evaluation)