コード例 #1
0
ファイル: main.py プロジェクト: mcass19/decision_tree
id3 = Id3()
attributes_aux = attributes.copy()
tree = id3.generate_tree(data_set, attributes, attributes_aux)

print('Generó árbol')

print('\n')

print('Árbol ID3 generado:')
tree.print(0)

print('\n')

evaluation = Evaluation()
cant_classified, cant_not_classified = evaluation.evaluate_tree(
    tree, data_set_test)
print('De las {} instancias tomadas para evaluar:'.format(cant_classified +
                                                          cant_not_classified))
print('\t -> {} instancias clasificaron correctamente'.format(cant_classified))
print('\t -> {} instancias clasificaron incorrectamente'.format(
    cant_not_classified))

print('\n')

print('Árboles de clases generados: ')

classes_trees = []

for label in data_set.target_values():
    data_set_class = data_set.data_set_class(label)
    attributes_aux = attributes.copy()