oracle-ads 2.13.9rc0__py3-none-any.whl → 2.13.10rc0__py3-none-any.whl

ads/dataset/sampled_dataset.py

@@ -0,0 +1,1050 @@
+#!/usr/bin/env python
+# -*- coding: utf-8; -*-
+
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+import matplotlib
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+import itertools
+
+from matplotlib.patches import BoxStyle
+
+from ads.dataset.label_encoder import DataFrameLabelEncoder
+from sklearn.feature_selection import SelectKBest
+from sklearn.feature_selection import chi2
+from sklearn.preprocessing import MinMaxScaler
+
+import matplotlib.font_manager
+from ads.common import utils, logger
+from ads.dataset.helper import (
+    fix_column_names,
+    convert_columns,
+    get_feature_type,
+    convert_to_html,
+)
+from ads.dataset.plot import Plotting
+from ads.dataset.progress import DummyProgressBar
+from ads.dataset.timeseries import Timeseries
+from ads.type_discovery.type_discovery_driver import TypeDiscoveryDriver
+from ads.type_discovery.typed_feature import (
+    DateTimeTypedFeature,
+    ContinuousTypedFeature,
+    GISTypedFeature,
+    ConstantTypedFeature,
+    CreditCardTypedFeature,
+    ZipcodeTypedFeature,
+    PhoneNumberTypedFeature,
+    OrdinalTypedFeature,
+    CategoricalTypedFeature,
+    DocumentTypedFeature,
+    AddressTypedFeature,
+)
+from ads.common.decorator.runtime_dependency import (
+    runtime_dependency,
+    OptionalDependency,
+)
+
+NATURAL_EARTH_DATASET = "naturalearth_lowres"
+
+
+class PandasDataset(object):
+    """
+    This class provides APIs that can work on a sampled dataset.
+    """
+
+    def __init__(
+        self,
+        sampled_df,
+        type_discovery=True,
+        types={},
+        metadata=None,
+        progress=DummyProgressBar(),
+    ):
+        self.client = None
+        self.sampled_df = fix_column_names(sampled_df)
+        self.correlation = None
+        self.feature_dist_html_dict = {}
+        self.feature_types = metadata if metadata is not None else {}
+        self.world = None
+
+        self.numeric_columns = self.sampled_df.select_dtypes(
+            utils.numeric_pandas_dtypes()
+        ).columns.values
+
+        # run type discovery
+        if len(self.feature_types) == 0:
+            if len(types) != 0:
+                # update feature types as it is for the types given by user
+                self._update_feature_types(types.keys())
+            if type_discovery:
+                try:
+                    #
+                    # perform type-discovery
+                    #
+                    progress.update("Running data type discovery")
+                    tdd = TypeDiscoveryDriver()
+                    self.feature_types = {
+                        col_name: tdd.discover(col_name, col_vals)
+                        for col_name, col_vals in self.sampled_df.items()
+                        if col_name not in types
+                    }
+                except Exception as e:
+                    print(
+                        f"An error occured while performing typed discovery on this dataset. Try running again with "
+                        f"`type_discovery=False`"
+                    )
+                    raise e
+
+        # convert dataframe columns to the data types discovered.
+        self.sampled_df = convert_columns(self.sampled_df, self.feature_types)
+
+        # update feature types for rest of the columns - no type discovery, does not include user overrides
+        self._update_feature_types(
+            set(self.sampled_df.columns.values) - set(self.feature_types.keys())
+        )
+        self.sampled_df = self.sampled_df.reset_index(drop=True)
+
+    def _find_feature_subset(self, df, target_name, include_n_features=32):
+        if len(df.columns) <= include_n_features:
+            return self.sampled_df
+        else:
+            try:
+                y = df[target_name]
+                X = df.drop(columns=[target_name])
+                X_columns = X.columns
+
+                X = X.fillna(X.mode().iloc[0])
+                X = DataFrameLabelEncoder().fit_transform(X)
+                X = MinMaxScaler().fit_transform(X)
+
+                from sklearn.impute import SimpleImputer
+
+                imp_most_freq = SimpleImputer(strategy="most_frequent")
+                X = imp_most_freq.fit_transform(X)
+
+                est = SelectKBest(score_func=chi2, k=include_n_features)
+                est.fit_transform(X, y)
+
+                subset_features = [self.target.name] + list(
+                    itertools.compress(X_columns, est.get_support())
+                )
+
+                return self.sampled_df.filter(subset_features, axis=1)
+
+            except Exception as e:
+                print("_find_feature_subset failed: ", str(e))
+                return pd.DataFrame()
+
+    def _update_multiple_outputs(self, out, msg):
+        if isinstance(out, (list, tuple)):
+            for o in out:
+                o.value = msg
+        else:
+            self._update_multiple_outputs([out], msg)
+
+    def _calculate_dataset_statistics(self, is_wide_dataset, out):
+        #
+        # first the missing values for non-wide datasets
+        #
+        df_missing = pd.DataFrame()
+        df_skew = pd.DataFrame()
+
+        if is_wide_dataset:
+            df_missing = pd.DataFrame()  # empty dataframe when not calculating
+            df_skew = pd.DataFrame()  # empty dataframe when not calculating
+        else:
+            #
+            # count missing values
+            #
+            d = {column_name: np.nan for column_name in self.df.columns}  # default
+            for column_name in self.df.columns:
+                self._update_multiple_outputs(
+                    out, f"calculating missing values (<code>{column_name}</code>)..."
+                )
+                d[column_name] = self.df[column_name].isna().sum()
+
+            df_missing = pd.DataFrame.from_dict(d, orient="index", columns=["missing"])
+
+            #
+            # calculate skew
+            #
+            d = {column_name: np.nan for column_name in self.df.columns}  # default
+            for column_name in self.numeric_columns:
+                self._update_multiple_outputs(
+                    out, f"calculating skew (<code>{column_name}</code>)..."
+                )
+                if len(self.df[column_name].dropna()) > 0:
+                    d[column_name] = self.df[column_name].dropna().skew()
+                else:
+                    d[column_name] = np.nan
+            df_skew = pd.DataFrame.from_dict(d, orient="index", columns=["skew"])
+            self._update_multiple_outputs(out, "calculating dataset statistics...")
+            for col in self.df.columns:
+                if df_missing["missing"][col] == len(self.df[col]):
+                    self.df[col] = self.df[col].astype("object")
+
+        #
+        # compute descriptive dataset statistics
+        df_stats = self.df.describe(include="all").round(2)
+
+        self._update_multiple_outputs(out, "transpose dataset statistics...")
+        df_stats = df_stats.transpose()
+
+        self._update_multiple_outputs(out, "finalizing dataset statistics...")
+        df_stats = df_stats.fillna("")
+
+        #
+        # we join the stats with missing df if we computed that
+        #
+        self._update_multiple_outputs(
+            out, "Assembling statistics into single result..."
+        )
+        if not df_missing.empty:
+            df_stats = df_stats.join(df_missing).fillna("")
+        if not df_skew.empty:
+            df_stats = df_stats.join(df_skew).fillna("")
+
+        return df_stats
+
+    def _generate_features_html(
+        self, is_wide_dataset, n_features, df_stats, visualizations_follow
+    ):
+        html = utils.get_bootstrap_styles()
+
+        if is_wide_dataset:
+            html += """<p>The dataset has too many columns ({:,}) to
+                efficiently show feature visualizations, instead only showing table of
+                statistics</p>""".format(
+                n_features
+            )
+
+        html += "<p><b>&#x2022; Note</b> these are computed on the entire dataset.</p>"
+        html += "<hr>"
+        html += (
+            df_stats.replace(np.nan, "")
+            .style.set_table_styles(utils.get_dataframe_styles(max_width=125))
+            .set_table_attributes("class=table")
+            .format(
+                lambda x: "{:.8g}".format(x)
+                if ("float" in str(type(x))) or ("int" in str(type(x)))
+                else x
+            )
+            .to_html()
+        )
+
+        if visualizations_follow:
+            html += "<br><hr><h2>Feature Visualizations...</h2>"
+
+        return html
+
+    def _generate_warnings_html(
+        self, is_wide_dataset, n_rows, n_features, df_stats, out, accordion
+    ):
+        #
+        # create the "Warnings" accordion section:
+        #  - show high cardinal categoricals
+        #  - show high missing values
+        #  - large number of zeros (not for wide datasets - slow to compute)
+        #
+        #
+
+        accum = []
+
+        ignored_feature_types = ["document"]
+
+        # more than 5% missing is a warning
+        #
+
+        if "missing" in df_stats.columns:
+            df_missing = df_stats[["missing"]][df_stats["missing"] != ""]
+            if not df_missing.empty:
+                # ignore document types
+                for column_name, missing in df_missing.iterrows():
+                    if (
+                        self.feature_types[column_name]["type"]
+                        not in ignored_feature_types
+                    ):
+                        missing_values = float(missing[0])
+                        missing_pct = 100 * missing_values / n_rows
+                        if missing_pct >= 5:
+                            accum.append(
+                                {
+                                    "label": "missing",
+                                    "message": f"<code>{column_name}</code> has {missing_values} ({missing_pct:.1f}%) missing values. Consider remove the column or replace null values.",
+                                }
+                            )
+                        if missing_pct == 100:
+                            accum.append(
+                                {
+                                    "label": "missing",
+                                    "message": f"<code>{column_name}</code> is excluded from correlation computation due to {missing_values} ({missing_pct:.1f}%) missing values.",
+                                }
+                            )
+
+        # abs skew > 20 skewness warning issues
+        #
+
+        if "skew" in df_stats.columns:
+            df_skew = df_stats[["skew"]][df_stats["skew"] != ""]
+            if not df_skew.empty:
+                # ignore document types
+                for column_name, skew in df_skew.iterrows():
+                    if (
+                        self.feature_types[column_name]["type"]
+                        not in ignored_feature_types
+                    ):
+                        skewness = float(skew[0])
+                        if abs(skewness) >= 20:
+                            accum.append(
+                                {
+                                    "label": "skew",
+                                    "message": f"<code>{column_name}</code> has skew of {skewness:.3f}",
+                                }
+                            )
+
+        # high cardinality (> 15 unique values) is a warning
+        #
+
+        if "unique" in df_stats.columns:
+            for column_name, count in df_stats[["unique"]][
+                df_stats["unique"] != ""
+            ].iterrows():
+                # ignore document types
+                if self.feature_types[column_name]["type"] not in ignored_feature_types:
+                    out.value = (
+                        f"Analyzing cadinalities (<code>{column_name}</code>)..."
+                    )
+                    unique = self.df[column_name].unique().shape[0]
+                    if unique == n_rows:
+                        accum.append(
+                            {
+                                "label": "high-cardinality",
+                                "message": f"<code>{column_name}</code> has a high cardinality: every value is distinct",
+                            }
+                        )
+                    elif unique > 15:
+                        accum.append(
+                            {
+                                "label": "high-cardinality",
+                                "message": f"<code>{column_name}</code> has a high cardinality: {unique} distinct values",
+                            }
+                        )
+
+        if not is_wide_dataset:
+            # more than 10% zeros is a warning
+            if "min" in df_stats.columns:
+                for column_name, count in df_stats[["min"]][
+                    df_stats["min"] != ""
+                ].iterrows():
+                    if (
+                        self.feature_types[column_name]["type"]
+                        not in ignored_feature_types
+                    ):
+                        out.value = f"Analyzing zeros (<code>{column_name}</code>)..."
+                        # we've filtered on only columns that have a min value of 0 for speed
+                        zeros = self.df[self.df[column_name] == 0].shape[0]
+                        zeros_pct = 100 * zeros / n_rows
+                        if zeros_pct >= 10:
+                            accum.append(
+                                {
+                                    "label": "zeros",
+                                    "message": f"<code>{column_name}</code> has {zeros} ({zeros_pct:.2f}%) zeros)",
+                                }
+                            )
+
+        #
+        # collect the warnings into an HTML presentation
+        #
+
+        out.value = "Assembling results..."
+
+        if accum:
+            html = utils.get_bootstrap_styles()
+
+            html += utils.highlight_text(f"{len(accum)} WARNING(S) found")
+
+            accordion.set_title(3, f"Warnings ({len(accum)})")
+
+            html += "<hr>"
+
+            tr_rows = [
+                f"""
+                <tr style="border-top:0">
+                       <td>
+                        {feature_warning['message']}
+                       <td>
+                        <span class="label label-{feature_warning['label']}">{feature_warning['label']}</span>
+                       </td>
+                </tr>
+            """.strip()
+                for feature_warning in accum
+            ]
+
+            html += """
+                <table style="width: 100%; max-width: 100%;">
+                <tbody>
+                {}
+                </tbody>
+                </table>""".format(
+                "\n".join(tr_rows)
+            )
+
+            return html
+
+        else:
+            return "<h3>No Feature warnings found</h3>"
+
+    def summary(self, feature_name=None):
+        """
+        Display list of features & their datatypes.
+        Shows the column name and the feature's meta_data if given a specific feature name.
+
+        Parameters
+        ----------
+        date_col: str
+            The name of the feature
+
+        Returns
+        -------
+        dict
+            a dictionary that contains requested information
+        """
+
+        feature_n_datatype = {}
+        list_of_dfs = []
+
+        if feature_name is None:
+            feature_n_datatype = {
+                col: self.feature_types[col].type
+                + "/"
+                + self.feature_types[col].low_level_type
+                for col in self.sampled_df.columns
+            }
+            df = pd.DataFrame(
+                feature_n_datatype.items(), columns=["Feature", "Datatype"]
+            )
+        else:
+            if isinstance(feature_name, (list, tuple, pd.core.indexes.base.Index)):
+                feature_names = list(feature_name)
+            else:
+                feature_names = [feature_name]
+
+            for col in feature_names:
+                if col in self.sampled_df.columns:
+                    feature_n_datatype[col] = {
+                        k: v
+                        for k, v in self.feature_types[col].meta_data.items()
+                        if k not in ["internal", "feature_name"]
+                    }
+                    new_dict = utils.flatten(feature_n_datatype[col])
+                    tmp_df = pd.DataFrame.from_dict(
+                        new_dict,
+                        orient="index",
+                        columns=[feature_names[feature_names.index(str(col))]],
+                    )
+                    list_of_dfs.append(tmp_df)
+                else:
+                    feature_n_datatype[col] = None
+
+            df = pd.concat(list_of_dfs, axis=1).transpose().fillna("-")
+
+            # get all the unique types from df
+            new_list_dfs = []
+            for t in df.type.unique():
+                new_list_dfs.append(df[df["type"] == t])
+            df = pd.concat(new_list_dfs)
+
+            # reorder columns in df
+
+            # get a list of columns
+            cols = list(df)
+            # move the column to head of list using index, pop and insert
+            cols.insert(0, cols.pop(cols.index("low_level_type")))
+            cols.insert(0, cols.pop(cols.index("type")))
+            df = df.loc[:, cols]
+
+        return df
+
+    def timeseries(self, date_col):
+        """
+        Supports any plotting operations where x=datetime.
+
+        Parameters
+        ----------
+        date_col: str
+            The name of the feature to plot
+
+        Returns
+        -------
+        func
+            a plotting object that contains a date column and dataframe
+        """
+
+        if date_col in self.feature_types and isinstance(
+            self.feature_types[date_col], DateTimeTypedFeature
+        ):
+            return Timeseries(date_col, self.sampled_df)
+        else:
+            raise ValueError("Not a date time column.")
+
+    def plot(
+        self, x, y=None, plot_type="infer", yscale=None, verbose=True, sample_size=0
+    ):
+        """
+        Supports plotting feature distribution, and relationship between features.
+
+        Parameters
+        ----------
+        x: str
+            The name of the feature to plot
+        y: str, optional
+            Name of the feature to plot against x
+        plot_type: str, default: infer
+            Override the inferred plot type for certain combinations of the data types of x and y.
+            By default, the best plot type is inferred based on x and y data types.
+            Valid values:
+
+            - box_plot - discrete feature vs continuous feature. Draw a box plot to show
+              distributions with respect to categories,
+            - scatter - continuous feature vs continuous feature. Draw a scatter plot
+              with possibility of several semantic groupings.
+
+        yscale : str, optional
+            One of {"linear", "log", "symlog", "logit"}.
+            The y axis scale type to apply. Can be used when either x or y is an ordinal feature.
+        verbose: bool, default True
+            Displays Note/Tips if True
+        """
+        sample_size = int(sample_size)
+        min_sample_size = 10000
+        if sample_size == 0:
+            sub_samp_size = len(self.sampled_df)
+            sub_samp_df = self.sampled_df
+        else:
+            sub_samp_size = max(min(sample_size, len(self.sampled_df)), min_sample_size)
+            sub_samp_df = self.sampled_df.sample(n=sub_samp_size)
+        plot = Plotting(
+            sub_samp_df, self.feature_types, x, y=y, plot_type=plot_type, yscale=yscale
+        )
+        if verbose:
+            if len(self.df) != sub_samp_size:
+                logger.info(f"Downsampling from dataset for graphing.")
+        return plot
+
+    @runtime_dependency(module="geopandas", install_from=OptionalDependency.GEO)
+    def plot_gis_scatter(self, lon="longitude", lat="latitude", ax=None):
+        """
+        Supports plotting Choropleth maps
+
+        Parameters
+        ----------
+        df: pandas dataframe
+            The dataframe to plot
+        x: str
+            The name of the feature to plot, usually the longitude
+        y: str
+            THe name of the feature to plot, usually the latitude
+        """
+        if lon in self.sampled_df.columns and lat in self.sampled_df.columns:
+            if ax is None:
+                fig, ax = plt.subplots(1, figsize=(10, 10))
+            gdf = geopandas.GeoDataFrame(
+                self.sampled_df,
+                geometry=geopandas.points_from_xy(
+                    self.sampled_df[lon], self.sampled_df[lat]
+                ),
+            )
+            world = geopandas.read_file(
+                geopandas.datasets.get_path(NATURAL_EARTH_DATASET)
+            )
+            ax1 = world.plot(ax=ax, color="lightgrey", linewidth=0.5, edgecolor="white")
+            gdf.plot(ax=ax1, color="blue", markersize=10)
+
+        else:
+            if len(self.sampled_df.columns) > 0:
+                logger.info(
+                    "The available latitude and longitude columns are: "
+                    + ", ".join(self.sampled_df.columns)
+                    + "."
+                )
+            else:
+                logger.info("There are no latitude and longitude columns available.")
+
+    """
+    Internal methods
+    """
+
+    def _update_feature_types(self, columns):
+        # Build feature types for columns which are not type discovered, by using the inferred type as it is
+        for column in columns:
+            self.feature_types[column] = get_feature_type(
+                column, self.sampled_df[column]
+            )
+
+    @runtime_dependency(module="geopandas", install_from=OptionalDependency.GEO)
+    def _visualize_feature_distribution(self, html_widget):
+        """
+        This function is called once per dataset to generate html for feature distribution plots.
+        """
+        if len(self.feature_dist_html_dict) > 0:
+            return self.feature_dist_html_dict
+
+        feature_dist_html = ""
+        figsize = (6.5, 2)
+
+        props = {
+            "boxstyle": BoxStyle("Round", pad=0),
+            "facecolor": "white",
+            "linestyle": "solid",
+            "linewidth": 0,
+            "edgecolor": "white",
+        }
+
+        font = {"size": 10}
+
+        matplotlib.rc("font", **font)
+
+        red_square = dict(markerfacecolor="r", marker="s")
+
+        blues = [
+            "#AED6F1",
+            "#85C1E9",
+            "#5DADE2",
+            "#3498DB",
+            "#2E86C1",
+            "#2874A6",
+            "#1B4F72",
+        ]
+        bright_colors = ["red", "green", "blue", "yellow", "green"]
+
+        for col_index, col in enumerate(self.sampled_df.columns):
+            feature_metadata = self.feature_types[col].meta_data
+            text = "\n".join(
+                [
+                    f"{col}\n",
+                    "  - type: {} ({})".format(
+                        self.feature_types[col].type,
+                        self.feature_types[col].low_level_type,
+                    ),
+                    "  - missing_percentage: {:.1f}%".format(
+                        self.feature_types[col].missing_percentage
+                    ),
+                ]
+            )
+
+            fig, ax = PandasDataset._init_fig_ax(figsize)
+
+            if isinstance(self.feature_types[col], ContinuousTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                self.sampled_df[col].plot(
+                    kind="box",
+                    vert=False,
+                    flierprops=red_square,
+                    ax=ax,
+                    figsize=figsize,
+                )
+
+            elif isinstance(self.feature_types[col], DateTimeTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                from matplotlib.dates import AutoDateFormatter, AutoDateLocator
+
+                loc = AutoDateLocator(interval_multiples=False)
+                ax.xaxis.set_major_locator(loc)
+                ax.xaxis.set_major_formatter(AutoDateFormatter(loc))
+                self.sampled_df[col].hist(
+                    bins=50,
+                    grid=False,
+                    xrot=45,
+                    ax=ax,
+                    rwidth=0.95,
+                    color=blues[-1],
+                    figsize=figsize,
+                )
+
+            elif isinstance(self.feature_types[col], GISTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                df = pd.DataFrame(
+                    feature_metadata["internal"]["sample"], columns=["lat", "lon"]
+                )
+                gdf = geopandas.GeoDataFrame(
+                    df, geometry=geopandas.points_from_xy(df["lon"], df["lat"])
+                )
+
+                if not self.world:
+                    self.world = geopandas.read_file(
+                        geopandas.datasets.get_path(NATURAL_EARTH_DATASET)
+                    )
+
+                self.world.plot(
+                    ax=ax, color="lightgrey", linewidth=0.5, edgecolor="white"
+                )
+                gdf.plot(ax=ax, color="blue", markersize=10)
+                ax.set_aspect("auto")
+
+            elif (
+                isinstance(self.feature_types[col], ConstantTypedFeature)
+                and feature_metadata["missing_percentage"] < 100
+            ):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                pd.Series(
+                    feature_metadata["internal"]["counts"],
+                    name=col,
+                    index=feature_metadata["internal"]["counts"].keys(),
+                ).plot(kind="barh", ax=ax, width=0.95, figsize=figsize, color=["black"])
+
+            elif isinstance(self.feature_types[col], CreditCardTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                sorted_by_value = sorted(
+                    feature_metadata["internal"]["counts"],
+                    key=feature_metadata["internal"]["counts"].get,
+                    reverse=True,
+                )
+                pd.Series(
+                    feature_metadata["internal"]["counts"],
+                    name=col,
+                    index=sorted_by_value,
+                ).plot(kind="bar", ax=ax, width=0.95, figsize=figsize, color=blues)
+
+            elif isinstance(self.feature_types[col], ZipcodeTypedFeature):
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                pd.Series(
+                    feature_metadata["internal"]["histogram"],
+                    name=col,
+                    index=feature_metadata["internal"]["histogram"].keys(),
+                ).plot(kind="bar", ax=ax, figsize=figsize, color=blues)
+
+            elif isinstance(self.feature_types[col], PhoneNumberTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                pd.Series(
+                    feature_metadata["internal"]["counts"],
+                    name=col,
+                    index=feature_metadata["internal"]["counts"].keys(),
+                ).plot(kind="bar", ax=ax, figsize=figsize, color=blues)
+
+            elif isinstance(self.feature_types[col], OrdinalTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+                high_cardinality = feature_metadata["internal"]["high_cardinality"]
+                very_high_cardinality = feature_metadata["internal"][
+                    "very_high_cardinality"
+                ]
+
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+
+                if very_high_cardinality:
+                    addrtext = "Samples:\n\n"
+                    addrtext += ", ".join(
+                        feature_metadata["internal"]["counts"]
+                        .keys()
+                        .astype(str)
+                        .to_list()[:6]
+                    )
+
+                    ax.text(
+                        0.05,
+                        0.95,
+                        addrtext,
+                        transform=ax.transAxes,
+                        fontsize=10,
+                        verticalalignment="top",
+                        bbox=props,
+                    )
+                    ax.axis("off")
+
+                else:
+                    if high_cardinality:
+                        text += (
+                            "\n\n  NOTE: plot has been capped\n  from %d to show only most\n  common top %d "
+                            "categories"
+                            % (
+                                feature_metadata["internal"]["unique"],
+                                len(feature_metadata["internal"]["counts"].keys()),
+                            )
+                        )
+
+                    if feature_metadata["internal"]["unique"] < 24:
+                        pd.Series(
+                            feature_metadata["internal"]["counts"],
+                            name=col,
+                            index=feature_metadata["internal"]["counts"].keys(),
+                        ).plot(
+                            kind="bar",
+                            ax=ax,
+                            width=0.90,
+                            color=blues[-1],
+                            figsize=figsize,
+                        )
+                    else:
+                        self.sampled_df[col].plot(
+                            kind="hist",
+                            grid=False,
+                            rwidth=0.95,
+                            ax=ax,
+                            color=blues[-1],
+                            figsize=figsize,
+                        )
+
+            elif isinstance(self.feature_types[col], CategoricalTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type, feature_metadata["stats"]
+                )
+
+                high_cardinality = feature_metadata["internal"]["high_cardinality"]
+                very_high_cardinality = feature_metadata["internal"][
+                    "very_high_cardinality"
+                ]
+
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+
+                if very_high_cardinality:
+                    # grab the first few examples as strings
+                    addrtext = "Samples:\n\n"
+                    addrtext += "\n".join(
+                        [
+                            utils.ellipsis_strings(x, 65)
+                            for x in feature_metadata["internal"]["counts"]
+                            .keys()
+                            .astype(str)
+                            .to_list()[:3]
+                        ]
+                    )
+                    ax.text(
+                        0.05,
+                        0.95,
+                        addrtext,
+                        transform=ax.transAxes,
+                        fontsize=10,
+                        verticalalignment="top",
+                        bbox=props,
+                    )
+                    ax.axis("off")
+
+                else:
+                    text += PandasDataset._format_stats(
+                        self.feature_types[col].type, feature_metadata["stats"]
+                    )
+                    if high_cardinality:
+                        text += (
+                            "\n\n  NOTE: plot has been capped\n  to show only most\n  common top %d categories"
+                            % (len(feature_metadata["internal"]["counts"].keys()))
+                        )
+
+                    if feature_metadata["internal"]["unique"] == 2:
+                        #
+                        # binary
+                        #
+                        count_series = pd.Series(
+                            feature_metadata["internal"]["counts"],
+                            name=col,
+                            index=feature_metadata["internal"]["counts"].keys(),
+                        ).astype(float)
+                        ax1 = count_series.plot(
+                            kind="barh",
+                            ax=ax,
+                            width=0.95,
+                            figsize=figsize,
+                            color=[blues[0], blues[-1]],
+                        )
+                        # x_labels = utils.ellipsis_strings(feature_metadata['internal']['counts'].keys().astype(str))
+                        # ax1.set_xticklabels(x_labels)
+                    else:
+                        #
+                        # multiclass, potentially high cardinality
+                        #
+
+                        ax1 = pd.Series(
+                            feature_metadata["internal"]["counts"],
+                            name=col,
+                            index=feature_metadata["internal"]["counts"].keys(),
+                        ).plot(
+                            kind="bar", ax=ax, width=0.95, color=blues, figsize=figsize
+                        )
+
+                        x_labels = utils.ellipsis_strings(
+                            feature_metadata["internal"]["counts"].keys()
+                        )
+                        ax1.set_xticklabels(x_labels)
+
+            elif isinstance(
+                self.feature_types[col], DocumentTypedFeature
+            ) or isinstance(self.feature_types[col], AddressTypedFeature):
+                text += PandasDataset._format_stats(
+                    self.feature_types[col].type,
+                    {k: v for k, v in feature_metadata["stats"].items()},
+                )
+
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                )
+                if "word_frequencies" in feature_metadata["internal"]:
+                    word_freqs = feature_metadata["internal"]["word_frequencies"]
+                    stats = "\n".join(["  - word count: %d" % (len(word_freqs.keys()))])
+                    text = text + "\n"
+                    text += stats
+                    try:
+                        from wordcloud import WordCloud, STOPWORDS
+
+                        wordcloud = WordCloud(
+                            width=1000,
+                            height=int(1000 * (figsize[1] / figsize[0])),
+                            background_color="white",
+                            stopwords=set(STOPWORDS),
+                            max_words=50,
+                            max_font_size=75,
+                        ).fit_words(word_freqs)
+
+                        plt.imshow(wordcloud, interpolation="bilinear")
+                        plt.axis("off")
+                    except ModuleNotFoundError as e:
+                        utils._log_missing_module("wordcloud", OptionalDependency.TEXT)
+                        logger.info(
+                            "The text word cloud is not plotted due to missing dependency wordcloud."
+                        )
+
+            else:
+                ax.text(
+                    -1.1,
+                    1,
+                    text,
+                    transform=ax.transAxes,
+                    verticalalignment="top",
+                    bbox=props,
+                    weight="bold",
+                )
+                if feature_metadata["missing_percentage"] == 100:
+                    addrtext = "NOTE: plot has been disabled as all values in this column are missing."
+                else:
+                    addrtext = "NOTE: plot has been disabled,\nfor features of unknown type\nno visualization is available"
+                ax.text(
+                    0.05,
+                    0.95,
+                    addrtext,
+                    transform=ax.transAxes,
+                    fontsize=14,
+                    verticalalignment="top",
+                    bbox=props,
+                    weight="bold",
+                )
+                ax.axis("off")
+
+            self.feature_dist_html_dict[col] = convert_to_html(plt)
+            plt.close()
+
+            html_widget.value += self.feature_dist_html_dict[col]
+
+    @staticmethod
+    def _init_fig_ax(figsize, dpi=288):
+        fig, ax = plt.subplots(figsize=figsize, dpi=dpi)
+        fig.set(facecolor="white")
+        return fig, ax
+
+    @staticmethod
+    def _format_stats(feature_type_name, stats):
+        text = "\n  - %s statistics:" % (feature_type_name)
+        for k in list(stats.keys()):
+            if "percentage" in k:
+                text += "\n    - {}: {:.3f}%".format(k, stats[k])
+            elif isinstance(stats[k], (int, np.int64)) or (
+                isinstance(stats[k], float)
+                and not np.isnan(stats[k])
+                and stats[k] == int(stats[k])
+            ):
+                text += "\n    - {}: {:,}".format(k, int(stats[k]))
+            elif isinstance(stats[k], bool):
+                text += "\n    - {}: {s}".format(k, "yes" if stats[k] else "no")
+            elif isinstance(stats[k], (float, np.float64)):
+                text += "\n    - {}: {:.3f}".format(k, stats[k])
+            else:
+                text += "\n    - {}: {}".format(k, stats[k])
+
+        return text