restored explanationresults class

Author: RobGeada

src/trustyai/explainers/counterfactuals.py

@@ -5,7 +5,6 @@
 import matplotlib.pyplot as plt
 import matplotlib as mpl
 import pandas as pd
-from pandas.io.formats.style import Styler  # pylint: disable=unused-import
 import uuid as _uuid
 
 from trustyai import _default_initializer  # pylint: disable=unused-import
@@ -20,6 +19,8 @@
     Model,
 )
 
+from .explanation_results import ExplanationResults
+
 
 from trustyai.utils.data_conversions import (
     prediction_object_to_numpy,
@@ -50,7 +51,7 @@
 CounterfactualConfig = _CounterfactualConfig
 
 
-class CounterfactualResult:
+class CounterfactualResult(ExplanationResults):
     """Wraps Counterfactual results. This object is returned by the
     :class:`~CounterfactualExplainer`, and provides a variety of methods to visualize and interact
     with the results of the counterfactual explanation.

src/trustyai/explainers/explanation_results.py

@@ -8,6 +8,19 @@
 from pandas.io.formats.style import Styler
 
 
+# pylint: disable=too-few-public-methods
+class ExplanationResults(ABC):
+    """Abstract class for non-saliency visualisers"""
+
+    @abstractmethod
+    def as_dataframe(self) -> pd.DataFrame:
+        """Display explanation result as a dataframe"""
+
+    @abstractmethod
+    def as_html(self) -> Styler:
+        """Visualise the styled dataframe"""
+
+
 # pylint: disable=too-few-public-methods
 class SaliencyResults(ABC):
     """Abstract class for saliency visualisers"""