# pm4py.visualization.decisiontree package

## pm4py.visualization.decisiontree.visualizer module

PM4Py is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

PM4Py is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with PM4Py. If not, see <https://www.gnu.org/licenses/>.

class pm4py.visualization.decisiontree.visualizer.Variants(value)[source]

Bases: enum.Enum

An enumeration.

CLASSIC = <module 'pm4py.visualization.decisiontree.variants.classic' from 'C:\\Users\\berti\\pm4py-core\\pm4py\\visualization\\decisiontree\\variants\\classic.py'>
pm4py.visualization.decisiontree.visualizer.apply(clf: sklearn.tree._classes.DecisionTreeClassifier, feature_names: List[str], classes: List[str], parameters: Optional[Dict[Any, Any]] = None, variant=Variants.CLASSIC) graphviz.sources.Source[source]

Method to apply the visualization of the decision tree

Parameters
• clf – Decision tree

• feature_names – Names of the provided features

• classes – Names of the target classes

• parameters

Possible parameters of the algorithm, including:

Parameters.FORMAT -> Image format (pdf, svg, png …)

• variant

Variant of the algorithm:
• Variants.CLASSIC

Returns

GraphViz object

Return type

gviz

pm4py.visualization.decisiontree.visualizer.matplotlib_view(gviz: graphviz.sources.Source)[source]

Views the diagram using Matplotlib

Parameters

gviz – Graphviz

pm4py.visualization.decisiontree.visualizer.save(gviz: graphviz.sources.Source, output_file_path: str)[source]

Save the diagram

Parameters
• gviz – GraphViz diagram

• output_file_path – Path where the GraphViz output should be saved

pm4py.visualization.decisiontree.visualizer.view(gviz: graphviz.sources.Source)[source]

View the diagram

Parameters

gviz – GraphViz diagram