Adjust Bounding Box of saved svg file #316

lgi1sgm · 2024-03-08T08:53:54Z

Hey there,

I want to use the SVG file coming out of the DTreeViz Render, but the bounding box around it is huge, and when I embed it into html code the tree becomes tiny.

dtreeviz/dtreeviz/utils.py

Line 269 in 0a84c38

def save(self, filename):

how can I get a tight bounding box around the SVG file?

Example

#!/usr/bin/env python


# Imports
# -------

import os
import dtreeviz

from pathlib import Path
from tempfile import NamedTemporaryFile
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import make_blobs


# Create temp files
# -----------------

tempfile = Path(NamedTemporaryFile().name)
svg_temp = tempfile.with_suffix('.svg')
html_temp = tempfile.with_suffix('.html')

print(f"You can delete this files later:\n{svg_temp}\n{html_temp}")


# Create example decision tree
# ----------------------------

X, y = make_blobs(n_samples=10, n_features=2, centers=3)

d = DecisionTreeClassifier()
d.fit(X, y)


# Export Decision Tree to SVG
# ---------------------------

tree_model = dtreeviz.model(d, X_train=X, y_train=y)
tree_render = tree_model.view()
tree_render.save(str(svg_temp))


# Create a HTML for displaying the SVG
# ------------------------------------

html = f"""
<!doctype html>
<html>
<head>
<style>
body {{margin: auto;max-width: 42em;}}
</style>
</head>
<body>
<img src="{svg_temp}"/>
</body>
"""

with open(html_temp, 'w') as fid:
  fid.write(html)


# Show whats the issue
# --------------------

# Expected Behavior: Graphic should be full size
# Experienced Behavior: A small tree is displayed in the upper left corner of the browser.

os.system('start ' + str(svg_temp))


# Experienced Behavior: Small tree fills only left side.
# Expected Behavior: An image of the tree fills out the complete width.

os.system('start ' + str(html_temp))

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Adjust Bounding Box of saved svg file #316

Adjust Bounding Box of saved svg file #316

lgi1sgm commented Mar 8, 2024 •

edited

Loading

Adjust Bounding Box of saved svg file #316

Adjust Bounding Box of saved svg file #316

Comments

lgi1sgm commented Mar 8, 2024 • edited Loading

lgi1sgm commented Mar 8, 2024 •

edited

Loading