Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-16 16:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add color keyword to arrows of graph

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-04-05 12:10:12 +02:00
parent ab2d96fe94
commit d678901930
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
stree/Splitter.py
View File

@@ -163,8 +163,8 @@ class Snode:
f"classes={count_values[0]} samples={count_values[1]} " f"classes={count_values[0]} samples={count_values[1]} "
f'({sum(count_values[1])})" fontcolor=black];\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_up())} [color=black];\n"
output += f"N{id(self)} -> N{id(self.get_down())};\n" output += f"N{id(self)} -> N{id(self.get_down())} [color=black];\n"
return output return output
def __str__(self) -> str: def __str__(self) -> str: