fix scaled_crps for pandas (#74)

Nixtla · Apr 9, 2024 · e32b0a7 · e32b0a7
1 parent c4d6229
commit e32b0a7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/nbs/losses.ipynb b/nbs/losses.ipynb
@@ -2056,6 +2056,7 @@
     "    sizes = ufp.counts_by_id(df, id_col)\n",
     "    if isinstance(loss, pd.DataFrame):\n",
     "        loss = loss.set_index(id_col)\n",
+    "        sizes = sizes.set_index(id_col)\n",
     "        assert isinstance(df, pd.DataFrame)\n",
     "        norm = df[target_col].abs().groupby(df[id_col], observed=True).sum()\n",
     "        res = 2 * loss.mul(sizes['counts'], axis=0).div(norm + eps, axis=0)\n",

diff --git a/settings.ini b/settings.ini
@@ -1,7 +1,7 @@
 [DEFAULT]
 repo = utilsforecast
 lib_name = utilsforecast
-version = 0.1.2
+version = 0.1.3
 min_python = 3.8
 license = apache2
 black_formatting = True

diff --git a/utilsforecast/__init__.py b/utilsforecast/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.2"
+__version__ = "0.1.3"
diff --git a/utilsforecast/losses.py b/utilsforecast/losses.py
@@ -627,6 +627,7 @@ def scaled_crps(
     sizes = ufp.counts_by_id(df, id_col)
     if isinstance(loss, pd.DataFrame):
         loss = loss.set_index(id_col)
+        sizes = sizes.set_index(id_col)
         assert isinstance(df, pd.DataFrame)
         norm = df[target_col].abs().groupby(df[id_col], observed=True).sum()
         res = 2 * loss.mul(sizes["counts"], axis=0).div(norm + eps, axis=0)