module plotting.visualize

Short summary

module mlinsights.plotting.visualize

Helpers to visualize a pipeline.

source on GitHub

Functions

function

truncated documentation

_pipeline_info

Internal function to convert a pipeline into some graph.

pipeline2dot

Exports a scikit-learn pipeline to DOT language. See Visualize a scikit-learn pipeline for an example.

Documentation

Helpers to visualize a pipeline.

source on GitHub

mlinsights.plotting.visualize._pipeline_info(pipe, data, context)[source]

Internal function to convert a pipeline into some graph.

source on GitHub

mlinsights.plotting.visualize.pipeline2dot(pipe, data, **params)[source]

Exports a scikit-learn pipeline to DOT language. See Visualize a scikit-learn pipeline for an example.

Parameters
  • pipescikit-learn pipeline

  • data – training data as a dataframe or a numpy array, or just a list with the variable names

  • params – additional params to draw the graph

Returns

string

Default options for the graph are:

options = {
    'orientation': 'portrait',
    'ranksep': '0.25',
    'nodesep': '0.05',
    'width': '0.5',
    'height': '0.1',
}

source on GitHub