From d67890193081786e057d7c84d12f15365044c350 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Tue, 5 Apr 2022 12:10:12 +0200 Subject: [PATCH] Add color keyword to arrows of graph --- stree/Splitter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stree/Splitter.py b/stree/Splitter.py index 4ef1f58..da58b07 100644 --- a/stree/Splitter.py +++ b/stree/Splitter.py @@ -163,8 +163,8 @@ class Snode: f"classes={count_values[0]} samples={count_values[1]} " 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" + output += f"N{id(self)} -> N{id(self.get_up())} [color=black];\n" + output += f"N{id(self)} -> N{id(self.get_down())} [color=black];\n" return output def __str__(self) -> str: