Databricks View Image

Question

I am running this in Databricks but the decision tree image will not display.

%pip install pydot
%pip install pydotplus

# Load libraries
from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from IPython.display import Image  
from sklearn import tree
import pydotplus

# Load data
iris = datasets.load_iris()
X = iris.data
y = iris.target

# Create DOT data
dot_data = tree.export_graphviz(clf, out_file=None, 
                                feature_names=iris.feature_names,  
                                class_names=iris.target_names)

# Draw graph
graph = pydotplus.graph_from_dot_data(dot_data)  

# Show graph
Image(graph.create_png())

I only get this message (no visual): Out[4]: <IPython.core.display.Image object>

I'm stumped. Thoughts?

Esben Eickhardt Esben Eickhardt · Accepted Answer · 2021-06-30T12:04:35

Databricks has the worst documentation, and their examples do not work at this time, so I had to come up with my own solution using PIL and Matplotlib.

Here is how I display images in Databricks in Python:

from PIL import Image
import matplotlib.pyplot as plt

def display_image(path, dpi=100):
    """
    Description:
        Displayes an image
    Inputs:
        path (str): File path
        dpi (int): Your monitors pixel density
    """
    img = Image.open(path)
    width, height = img.size
    plt.figure(figsize = (width/dpi,height/dpi))
    plt.imshow(img, interpolation='nearest', aspect='auto')

Databricks View Image

1 Answers