howso-visuals 2.5.6__py3-none-any.whl → 3.0.0__py3-none-any.whl

howso/visuals/tests/test_plot.py

@@ -53,9 +53,9 @@ def test_plot_interpretable_prediction_react(
             desired_conviction=5_000,
             num_cases_to_generate=20,
         )
-        generative_reacts = result["action"].loc[:, action_feature].values.tolist()
-        generative_reacts[0] = generative_reacts[0] + 0.2
-        generative_reacts[-1] = generative_reacts[-1] - 0.2
+        generative_reacts = result["action"]
+        generative_reacts.loc[0, action_feature] = generative_reacts.loc[0, action_feature] + 0.2
+        generative_reacts.loc[19, action_feature] = generative_reacts.loc[19, action_feature] - 0.2
     else:
         generative_reacts = None
 

howso/visuals/visuals.py

@@ -479,10 +479,11 @@ def plot_interpretable_prediction(
     react: "Reaction",
     *,
     actual_value: float | None = None,
-    generative_reacts: list[float] | None = None,
+    generative_reacts: pd.DataFrame | None = None,
     residual: float | None = None,
     secondary_yaxis_title: str = "Influence Weight",
-    title: str | None = None,
+    subplot_titles: list[str] | None = None,
+    title: str = "Interpretable Predictions",
     tooltip_features: Collection[str] | None = None,
     xaxis_title: str | None = None,
     yaxis_title: str = "Density",
@@ -495,10 +496,10 @@ def plot_interpretable_prediction(
     react : Reaction
         The reaction predicting the action feature(s) to visualize. If this contains more than one action feature,
         each will be given its own plot.
-    generative_reacts : list of float, optional
-        An optional list of values for the action feature to visualize. This will be used to visualize a
-        KDE plot to characterize the distribution of values around the predicted and actual values. If this is None,
-        the distribution of influential cases in the react will be used instead, if present.
+    generative_reacts : DataFrame, optional
+        Ann optional `DataFrame` of generative reactions which will be used to visualize a KDE around
+        the predicted value. If this is None, the distribution of influential cases in the react will
+        be used instead, if present.
     actual_value : float, optional
         The actual value for the point that was predicted. If this is None, only the predicted value will be
         visualized.
@@ -507,7 +508,9 @@ def plot_interpretable_prediction(
         predicted value. If this is None, no error bar will be displayed
     secondary_yaxis_title : str, default "Influence Weight"
         The title for the figure's secondary y axis.
-    title : str, optional
+    subplot_titles : list[str], optional
+        A list of titles to use for subplots. If this is None, titles will be auto-generated.
+    title : str, default "Interpretable Prediction"
         The title for the figure.
     tooltip_features : Collection[str], optional
         Additional features to include in the tooltip when hovering over an influential case in the plot.
@@ -521,31 +524,52 @@ def plot_interpretable_prediction(
     Figure or list of Figure
         The resultant `Plotly` figure(s).
     """
-    figures = []
     tooltip_features = list(tooltip_features) if tooltip_features is not None else []
+    action_features = react["action"].columns
+    num_rows = len(action_features)
+    subplot_titles = subplot_titles or [f"Interpretable Prediction for: {af}" for af in action_features]
+    case_hover_template = "<b>Value:</b> %{x}<br><b>Influence:</b> N/A"
+    fig = make_subplots(
+        rows=num_rows,
+        specs=[[{"secondary_y": True}]] * num_rows,
+        subplot_titles=subplot_titles,
+    )
 
-    for action_feature in react["action"].columns:
+    for row, action_feature in enumerate(action_features):
         predicted_value = react["action"][action_feature].iloc[0]
+        legend = "legend" if row == 0 else f"legend{row + 1}"
 
         influential_cases = react.get("details", {}).get("influential_cases")
         influential_cases = influential_cases[0] if influential_cases else None
 
         if generative_reacts is not None:
-            action_distribution = generative_reacts
-        else:
+            action_distribution = generative_reacts[action_feature]
+        elif influential_cases is not None:
             action_distribution = [c[action_feature] for c in influential_cases]
-        action_kde = gaussian_kde(action_distribution)
-        density_x = np.linspace(
-            min(action_distribution) * 0.6,
-            max(action_distribution) * 1.4,
-            len(action_distribution),
-        )
-        density_y = action_kde(density_x)
-        fig = make_subplots(specs=[[{"secondary_y": True}]])
-        fig.add_trace(
-            go.Scattergl(x=density_x, y=density_y, fill="tonexty", name="Distribution", hoverinfo="skip", mode="lines")
-        )
-        case_hover_template = "<b>Value:</b> %{x}<br><b>Influence:</b> N/A"
+        else:
+            action_distribution = None
+
+        if action_distribution is not None:
+            action_kde = gaussian_kde(action_distribution)
+            density_x = np.linspace(
+                min(action_distribution) * 0.6,
+                max(action_distribution) * 1.4,
+                len(action_distribution),
+            )
+            density_y = action_kde(density_x)
+            fig.add_trace(
+                go.Scattergl(
+                    x=density_x,
+                    y=density_y,
+                    fill="tonexty",
+                    name="Distribution",
+                    hoverinfo="skip",
+                    mode="lines",
+                    legend=legend,
+                ),
+                row=row + 1,
+                col=1,
+            )
         predicted_case_hover_template = case_hover_template
 
         if influential_cases is not None:
@@ -561,14 +585,17 @@ def plot_interpretable_prediction(
         fig.add_trace(
             go.Scattergl(
                 x=[predicted_value],
-                y=[max_inf_weight * 1.05],
+                y=[max_inf_weight * 1.05] if max_inf_weight else 1,
                 name="Predicted Value",
                 mode="markers",
                 marker={"size": 15, "color": "purple"},
                 hovertemplate=predicted_case_hover_template,
                 error_x=error_x,
+                legend=legend,
             ),
             secondary_y=True,
+            row=row + 1,
+            col=1,
         )
 
         if actual_value is not None:
@@ -576,19 +603,22 @@ def plot_interpretable_prediction(
             fig.add_trace(
                 go.Scattergl(
                     x=[actual_value],
-                    y=[max_inf_weight * 1.05],
+                    y=[max_inf_weight * 1.05] if max_inf_weight else 1,
                     name="Actual Value",
                     mode="markers",
                     marker=dict(size=15, color="orange"),
                     hovertemplate=case_hover_template,
+                    legend=legend,
                 ),
                 secondary_y=True,
+                row=row + 1,
+                col=1,
             )
 
         # Update axes, hover mode
-        fig.update_xaxes(title_text=xaxis_title or action_feature, autorange=True)
-        fig.update_yaxes(title_text=yaxis_title, color="blue", secondary_y=False)
-        fig.update_yaxes(title_text=secondary_yaxis_title, color="green", secondary_y=True)
+        fig.update_xaxes(title_text=xaxis_title or action_feature, autorange=True, row=row + 1, col=1)
+        fig.update_yaxes(title_text=yaxis_title, color="blue", secondary_y=False, row=row + 1, col=1)
+        fig.update_yaxes(title_text=secondary_yaxis_title, color="green", secondary_y=True, row=row + 1, col=1)
 
         if influential_cases is not None and len(influential_cases):
             inf_case_hover_template = (
@@ -621,19 +651,24 @@ def plot_interpretable_prediction(
                     customdata=inf_case_labels,
                     mode="markers",
                     marker=dict(color="green"),
+                    legend=legend,
                 ),
                 secondary_y=True,
+                row=row + 1,
+                col=1,
             )
 
-        if title is None:
-            title = f"Interpretable Prediction for: {action_feature}"
-        fig.update_layout(title=dict(text=title))
-
-        figures.append(fig)
+    # Increase height based on the number of rows
+    fig.update_layout(height=300 * num_rows)
+    # Place a legend to the right of each subplot.
+    for i, yaxis in enumerate(fig.select_yaxes(secondary_y=False), 1):
+        legend_name = f"legend{i}"
+        fig.update_layout({legend_name: dict(y=yaxis.domain[1], yanchor="top")}, showlegend=True)
+        fig.update_traces(row=i, legend=legend_name)
 
-    if len(figures) == 1:
-        return figures[0]
-    return figures
+    title = title or "Interpretable Predictions"
+    fig.update_layout(title=title)
+    return fig
 
 
 def plot_fairness_disparity(

{howso_visuals-2.5.6.dist-info → howso_visuals-3.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: howso-visuals
-Version: 2.5.6
+Version: 3.0.0
 Summary: Visualization utilities for use with Howso Engine.
 License: GNU AFFERO GENERAL PUBLIC LICENSE
            Version 3, 19 November 2007

{howso_visuals-2.5.6.dist-info → howso_visuals-3.0.0.dist-info}/RECORD

@@ -9,14 +9,14 @@ config/powershell/Helper-Functions.ps1,sha256=oLEunSYLr-zxtG9kqYuOGG8v3kSxTQEpqe
 howso/visuals/__init__.py,sha256=OL9LT4186YIXkEmM4qc21maRnfrtxBmFUHnalcKzGWA,507
 howso/visuals/colors.py,sha256=2ChjT4MUjA4ZqITR_K1j3fUKNRBd8M9iyJ4ZT10cNlw,1709
 howso/visuals/graph.py,sha256=fdZzYLwoBV8CeW_SwtcNkDh8xbIG_FlECSECxyBIr5M,8311
-howso/visuals/visuals.py,sha256=Aruqe2hUWCTcF2t_2HVbubvPF7RO7RG8RUxvr8EnMBg,33158
+howso/visuals/visuals.py,sha256=0jaM1f7Tzh6JVwrtNxQuVyhcOmgYoE-uu_-zzWef09w,34649
 howso/visuals/data/iris.csv,sha256=73iTb6rB42LgZwf89er-tEbuyjlvEAx6WywMPwt0GjU,2757
 howso/visuals/tests/conftest.py,sha256=BcvCVDWbwaQKSstzJzN9G48KoDnCgTtVdgYo2CJZTWM,1293
 howso/visuals/tests/test_graph.py,sha256=gfuJNpsNBLx2tDXDIR-ClbCZUSOQCmk8Dx2X-NhN2yM,1014
-howso/visuals/tests/test_plot.py,sha256=lOYYUty84LFFdaQCDcw-fMAicTqlHIwvNtuVch0RIEE,8121
-howso_visuals-2.5.6.dist-info/licenses/LICENSE-3RD-PARTY.txt,sha256=aosupVocnnJ4n99ICajoFbXaRnZstEbtxTuVaT6RWUQ,373165
-howso_visuals-2.5.6.dist-info/licenses/LICENSE.txt,sha256=2xqHuoHohba7gpcZZKtOICRjzeKsQANXG8WoV9V35KM,33893
-howso_visuals-2.5.6.dist-info/METADATA,sha256=ogmqaaDN4dIIimuzPcEvRjPSF5gXFl6DQjtHlvzH3E8,41051
-howso_visuals-2.5.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-howso_visuals-2.5.6.dist-info/top_level.txt,sha256=4ltSHx7mNsXczuoCPkz1irjq1x1JZir2QrX-ZwRNWXU,22
-howso_visuals-2.5.6.dist-info/RECORD,,
+howso/visuals/tests/test_plot.py,sha256=iVdEyuIRYEN23IZx4He3hjDvBwNOP9NsXqdrQPSYHaw,8162
+howso_visuals-3.0.0.dist-info/licenses/LICENSE-3RD-PARTY.txt,sha256=riZoS_SaicJQ2vten1_zvA3p0Y46ecEiJiDqZ6QoRCg,373165
+howso_visuals-3.0.0.dist-info/licenses/LICENSE.txt,sha256=2xqHuoHohba7gpcZZKtOICRjzeKsQANXG8WoV9V35KM,33893
+howso_visuals-3.0.0.dist-info/METADATA,sha256=0Y5RXJsuOvNyb0ISM8T0CuDYx-z7RHdlZOSWdh7bTTA,41051
+howso_visuals-3.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+howso_visuals-3.0.0.dist-info/top_level.txt,sha256=4ltSHx7mNsXczuoCPkz1irjq1x1JZir2QrX-ZwRNWXU,22
+howso_visuals-3.0.0.dist-info/RECORD,,

{howso_visuals-2.5.6.dist-info → howso_visuals-3.0.0.dist-info}/licenses/LICENSE-3RD-PARTY.txt

@@ -1,10 +1,10 @@
 Faker
-38.0.0
+38.2.0
 MIT License
 joke2k
 https://github.com/joke2k/faker
 Faker is a Python package that generates fake data for you.
-/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/faker-38.0.0.dist-info/licenses/LICENSE.txt
+/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/faker-38.2.0.dist-info/licenses/LICENSE.txt
 Copyright (c) 2012 Daniele Faraglia
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
@@ -297,12 +297,12 @@ OTHER DEALINGS IN THE SOFTWARE.
 
 
 click
-8.3.0
+8.3.1
 BSD-3-Clause
 UNKNOWN
 https://github.com/pallets/click/
 Composable command line interface toolkit
-/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/click-8.3.0.dist-info/licenses/LICENSE.txt
+/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/click-8.3.1.dist-info/licenses/LICENSE.txt
 Copyright 2014 Pallets
 
 Redistribution and use in source and binary forms, with or without
@@ -1232,12 +1232,12 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 
 narwhals
-2.11.0
+2.12.0
 MIT License
 Marco Gorelli <hello_narwhals@proton.me>
 https://github.com/narwhals-dev/narwhals
 Extremely lightweight compatibility layer between dataframe libraries
-/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/narwhals-2.11.0.dist-info/licenses/LICENSE.md
+/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/narwhals-2.12.0.dist-info/licenses/LICENSE.md
 MIT License
 
 Copyright (c) 2024, Marco Gorelli
@@ -5370,12 +5370,12 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 
 plotly
-6.4.0
+6.5.0
 MIT License
 Chris P <chris@plot.ly>
 https://plotly.com/python/
 An open-source interactive data visualization library for Python
-/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/plotly-6.4.0.dist-info/licenses/LICENSE.txt
+/home/runner/.pyenv/versions/3.13.9/lib/python3.13/site-packages/plotly-6.5.0.dist-info/licenses/LICENSE.txt
 MIT License
 
 Copyright (c) 2016-2024 Plotly Technologies Inc.