From ab2d96fe94ca8c56825d80c4291157ad978491c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Tue, 5 Apr 2022 12:00:41 +0200 Subject: [PATCH] Add fontcolor keyword to nodes of the tree --- stree/Splitter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stree/Splitter.py b/stree/Splitter.py index 4924721..4ef1f58 100644 --- a/stree/Splitter.py +++ b/stree/Splitter.py @@ -161,7 +161,7 @@ class Snode: output += ( f'N{id(self)} [label="#features={len(self._features)} ' f"classes={count_values[0]} samples={count_values[1]} " - f'({sum(count_values[1])})"];\n' + f'({sum(count_values[1])})" fontcolor=black];\n' ) output += f"N{id(self)} -> N{id(self.get_up())};\n" output += f"N{id(self)} -> N{id(self.get_down())};\n"