diff --git a/responsibleai_text/responsibleai_text/rai_text_insights/rai_text_insights.py b/responsibleai_text/responsibleai_text/rai_text_insights/rai_text_insights.py index 4e1be5e6ce..af179408e0 100644 --- a/responsibleai_text/responsibleai_text/rai_text_insights/rai_text_insights.py +++ b/responsibleai_text/responsibleai_text/rai_text_insights/rai_text_insights.py @@ -32,6 +32,8 @@ get_text_columns) from responsibleai_text.utils.genai_metrics.metrics import \ get_genai_metric_mean +from responsibleai_text.utils.genai_metrics.metrics import \ + get_genai_metric_mean module_logger = logging.getLogger(__name__) module_logger.setLevel(logging.INFO) @@ -154,9 +156,9 @@ def __init__(self, model: Any, test: pd.DataFrame, If not provided, and there is additional feature metadata, then an exception will be raised. :type text_column: str or list[str] - :param eval_model: The model to use for evaluation. - If not provided, then the model passed in the model parameter - will be used. + :param eval_model: The model to use for evaluation with AI-assisted + metrics. If not provided, then the model passed in the model + parameter will be used. :type eval_model: object """ # drop index as this can cause issues later like when copying diff --git a/responsibleai_text/responsibleai_text/utils/genai_metrics/__init__.py b/responsibleai_text/responsibleai_text/utils/genai_metrics/__init__.py new file mode 100644 index 0000000000..692faf31bd --- /dev/null +++ b/responsibleai_text/responsibleai_text/utils/genai_metrics/__init__.py @@ -0,0 +1,4 @@ +# Copyright (c) Microsoft Corporation +# Licensed under the MIT License. + +"""Contains the GenAI metrics.""" diff --git a/responsibleai_text/responsibleai_text/utils/genai_metrics/scripts/__init__.py b/responsibleai_text/responsibleai_text/utils/genai_metrics/scripts/__init__.py new file mode 100644 index 0000000000..461928c132 --- /dev/null +++ b/responsibleai_text/responsibleai_text/utils/genai_metrics/scripts/__init__.py @@ -0,0 +1,4 @@ +# Copyright (c) Microsoft Corporation +# Licensed under the MIT License. + +"""Contains the implementation of various metrics for GenAI."""