FileNotFoundError дерево принятия решений с CHAID - PullRequest
Новая
       30

FileNotFoundError дерево принятия решений с CHAID

0 голосов
/ 07 января 2020

Я использовал следующий код для получения дерева решений CHAID 

independent_variable_columns = ['gender', 'grade', 'no_renewals', 'complaint_count']
dep_variable = 'switch'
tree = Tree.from_pandas_df(
    df,
    dict(zip(independent_variable_columns, ['nominal'] * 38)),
    dep_variable,
    max_depth=2
)
tree.to_tree()
tree.render()

, но я получаю следующую ошибку 

  File "C:\Users\data\AppData\Local\Continuum\anaconda3\lib\site-packages\CHAID\graph.py", line 93, in <listcomp>
    [os.remove(file) for file in self.files]
FileNotFoundError: [WinError 3] The system cannot find the path specified: '/tmp/157840175993116164207458496094.png'

1 Ответ

1 голос
/ 07 января 2020

Указанная библиотека c не реализует лучшие практики, когда речь идет о генерации имен файлов и обработке временных файлов. Это приводит к тому, что их код по-разному ломается на Windows.

. Я оставил комментарий к аналогичной проблеме с той же root причиной , предлагая внести изменения в исправить эти проблемы.

Вы можете исправить это локально, запустив этот код: 

import os
from datetime import datetime
from tempfile import TemporaryDirectory

import plotly.io as pio
import colorlover as cl
from graphviz import Digraph

from CHAID import graph

def Graph_render(self, path, view):
    if path is None:
        path = os.path.join("trees", f"{datetime.now():%Y-%m-%d %H:%M:%S}.gv")
    with TemporaryDirectory() as self.tempdir:
        g = Digraph(
            format="png",
            graph_attr={"splines": "ortho"},
            node_attr={"shape": "plaintext", "labelloc": "b"},
        )
        for node in self.tree:
            image = self.bar_chart(node)
            g.node(str(node.node_id), image=image)
            if node.parent is not None:
                edge_label = f"     ({', '.join(map(str, node.choices))})     \n ")
                g.edge(str(node.parent), str(node.node_id), xlabel=edge_label)

        g.render(path, view=view)

def Graph_bar_chart(self, node):
    fig = {
        "data": [
            {
                "values": list(node.members.values()),
                "labels": list(node.members),
                "showlegend": node.node_id == 0,
                "domain": {"x": [0, 1], "y": [0.4, 1.0]},
                "hole": 0.4,
                "type": "pie",
                "marker_colors": cl.scales["5"]["qual"]["Set1"],
            },
        ],
        "layout": {
            "margin_t": 50,
            "annotations": [{"font_size": 18, "x": 0.5, "y": 0.5}, {"y": [0, 0.2]}],
        },
    }

    if not node.is_terminal:
        p = None if node.p is None else format(node.p, ".5f")
        score = None if node.score is None else format(node.score, ".2f")
        self.append_table(fig, [p, score, node.split.column])

    filename = os.path.join(self.tempdir, f"node-{node.node_id}.png")
    pio.write_image(fig, file=filename, format="png")
    return filename

graph.Graph.render = Graph_render
graph.Graph.bar_chart = Graph_bar_chart

Выше приведен рефакторинг двух методов из класса Graph в этой библиотеке, переключающихся на использование tempdir.TemporaryDirectory() для обработки временных файлов и очистки, как генерируются имена файлов. Я также позволил себе исправить некоторые незначительные проблемы в коде.

Я превратил это в запрос на извлечение для обновления самого проекта .

...