Ejemplo n.º 1
0
def dt_multiple_datasets(test_cases, args):
    print "DECISION TREE"
    print args

    for num_test in test_cases:
        if num_test == 1:
            decision_tree_ca(load.breast_cancer(), args)
            decision_tree_ca(load.car(), args)

        elif num_test == 2:
            decision_tree_ca(load.segmentation(), args)
            decision_tree_ca(load.iris(), args)
            decision_tree_ca(load.wine(), args)

        elif num_test == 3:
            decision_tree_ca(load.bank(), args)
            decision_tree_ca(load.lymphography(), args)
Ejemplo n.º 2
0
def clasifier_comparison(test_cases, args):
    print "Comparison of DECISION TREE, RANDOM FOREST"
    print args

    for num_test in test_cases:
        if num_test == 1:
            compare(load.breast_cancer(), args)
            compare(load.car(), args)

        elif num_test == 2:
            compare(load.segmentation(), args)
            compare(load.iris(), args)
            compare(load.wine(), args)

        if num_test == 3:
            compare(load.bank(), args)
            compare(load.lymphography(), args)