Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
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)