View source on GitHub |
Plots the model structure and its correlation with a dataset.
tfdf.model_plotter.plot_model(
model: InferenceCoreModel,
tree_idx: Optional[int] = 0,
max_depth: Optional[int] = 3
)
Args | |
---|---|
model
|
The model to plot. |
tree_idx
|
If the model is composed of multiple trees, only plot this tree. |
max_depth
|
Maximum depth to plot. If None, does not limit the depth. |
Returns | |
---|---|
Html representation of the tree. |