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

Add fontcolor keyword to nodes of the tree

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

2
stree/Splitter.py
View File

@@ -161,7 +161,7 @@ class Snode:
output += ( output += (
f'N{id(self)} [label="#features={len(self._features)} ' f'N{id(self)} [label="#features={len(self._features)} '
f"classes={count_values[0]} samples={count_values[1]} " 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_up())};\n"
output += f"N{id(self)} -> N{id(self.get_down())};\n" output += f"N{id(self)} -> N{id(self.get_down())};\n"