e8b2e17f4eec658f6a6d53486dbbe3eac48bb0fa,pyinterpret/core/global_interpretation/partial_dependence.py,PartialDependence,_plot_2d_2_binary_feature,#PartialDependence#Any#Any#Any#Any#Any#Any#,619

Before Change

            ax.set_xlabel(feature1)
            ax.set_ylabel("Predicted {}".format(class_name))
            handles, labels = ax.get_legend_handles_labels()
            ax.legend(handles, labels)
        return flatten([figure_list, axis_list])

    def _plot_2d_1_binary_feature_and_1_continuous(self, pdp, binary_feature,
                                                   non_binary_feature, pdp_metadata,

After Change

            //feature2 is columns
            //feature1 is index
            plot_data = pdp.set_index([feature1, feature2])[mean_col].unstack()
            plot_data.plot(ax=ax, color=COLORS)

            if with_variance:
                colors = cycle(COLORS)
                binary1_values = plot_data.index.values

Instances