Merge pull request #14 from knowlab/harry/adult_tutorial

Harry/adult tutorial

.github/workflows/pypi.yml

@@ -58,4 +58,4 @@ jobs:
           name: release-dists
           path: dist/
       - name: Publish release distributions to PyPI
-        uses: pypa/gh-action-pypi-publish@release/v1
+        uses: pypa/gh-action-pypi-publish@release/v1

.pre-commit-config.yaml

@@ -14,7 +14,7 @@ repos:
       - id: check-toml
       - id: check-added-large-files
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.6.4
+    rev: v0.6.5
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]

.vscode/settings.json

@@ -4,4 +4,4 @@
     ],
     "python.testing.unittestEnabled": false,
     "python.testing.pytestEnabled": true
-}
+}

daindex/core.py

@@ -102,7 +102,6 @@ def kde_estimate(
 
     kde = KernelDensity(bandwidth=bandwidth, kernel=kernel)
     kde.fit(X)
-
     return kde, bandwidth
 
 

daindex/util.py

@@ -131,7 +131,7 @@ def viz(
     g2_label: str,
     deterioration_label: str,
     allocation_label: str,
-    config: dict,
+    config: dict = {},
     decision_boundary: float = 0.5,
 ) -> None:
     """
@@ -181,3 +181,9 @@ def viz(
     plt.axvspan(decision_boundary, 1, facecolor="b", alpha=0.1)
 
     plt.legend(fontsize=font_size, loc="best")
+
+    return_dict = {"AUC": (a2 - a1) / a1}
+    if da1:
+        return_dict.update({"Decision AUC": (da2 - da1) / da1})
+
+    return return_dict

pyproject.toml

@@ -39,6 +39,7 @@ optional = true
 [tool.poetry.group.tutorials.dependencies]
 jupyter = "^1.1.1"
 notebook = "^7.2.2"
+aif360 = "^0.6.1"
 
 [tool.ruff]
 include = ["*.py", "*.pyi", "pyproject.toml", "*.ipynb"]