
234 Глава 8. Графовые алгоритмы имашинное обучение
featureNames=elds, categoryNames=[], classNames=["True", "False"],
lled=True, roundedCorners=True, roundLeaves=True)
with open("/tmp/rf.dot", "w") as le:
le.write(dot_string)
Затем можем сгенерировать визуальное представление этого файла,
выполнив следующую команду из терминала:
dot -Tpdf /tmp/rf.dot -o /tmp/rf.pdf
Результат выполнения этой команды представлен на рис. 8.12.
Рис. 8.12. Визуализация дерева решений
Представьте, что мы используем это дерево решений, чтобы предска-
зать, связана ли пара узлов, исходя из следующих признаков:
commonAuthors prefAttachment totalNeighbors
10 12 5
При ...