datfid 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

datfid/client.py

@@ -9,6 +9,40 @@ import gc
 import psutil
 import logging
 
+# for nice output
+class FitResult(SimpleNamespace):
+    _ROW4  = ["Estimate", "Standard Error", "T statistic", "P value"]
+    _PERF5 = ["R2 within", "R2 between", "R2 overall", "MSE", "MAE"]
+    
+    @property
+    def id(self):
+        """Return list of all individual IDs"""
+        return list(self.keys())
+
+    @staticmethod
+    def _df4(rows_list):
+        if not isinstance(rows_list, list):
+            return rows_list
+        return pd.DataFrame(rows_list, index=FitResult._ROW4[:len(rows_list)])
+
+    @staticmethod
+    def _df_perf(rows_list):
+        if not isinstance(rows_list, list):
+            return rows_list
+        return pd.DataFrame(rows_list, index=FitResult._PERF5[:len(rows_list)])
+
+    def __init__(self, **kwargs):
+        # Convert list→DataFrame for table-like fields
+        if "alpha" in kwargs:
+            kwargs["alpha"] = self._df4(kwargs["alpha"])
+        if "beta" in kwargs:
+            kwargs["beta"] = self._df4(kwargs["beta"])
+        if "Performance" in kwargs:
+            kwargs["Performance"] = self._df_perf(kwargs["Performance"])
+        if "df" in kwargs and isinstance(kwargs["df"], list):
+            kwargs["df"] = pd.DataFrame(kwargs["df"])
+        super().__init__(**kwargs)
+
 class DATFIDClient:
     def __init__(self, token: str):
         self.api_url = "https://datfid-org-datfid-sdk.hf.space/"
@@ -54,7 +88,7 @@ class DATFIDClient:
         current_features: Optional[list] = None,
         filter_by_significance: bool = False,
         meanvar_test: bool = False
-    ) -> SimpleNamespace:
+    ) -> FitResult:
         """
         Fit a model using the DATFID API.
         
@@ -99,8 +133,8 @@ class DATFIDClient:
             raise Exception(f"Model fit failed: {response.text}")
             
         result_dict = response.json()
-        return SimpleNamespace(**result_dict)
-
+        return FitResult(**result_dict)
+    
     def forecast_model(
         self,
         df_forecast: pd.DataFrame
@@ -156,7 +190,7 @@ class DATFIDClient:
         current_features: Optional[list] = None,
         filter_by_significance: bool = False,
         meanvar_test: bool = False
-    ) -> Dict[str, SimpleNamespace]:
+    ) -> Dict[str, FitResult]:
         """
         Fit a model individual by individual using the DATFID API.
         
@@ -202,7 +236,7 @@ class DATFIDClient:
             
         raw = response.json()  # { "<id>": { ... per-id result ... }, ... }
         # Wrap each per-id result into a SimpleNamespace for dot access:
-        result_per_id = {str(k): SimpleNamespace(**v) for k, v in raw.items()}
+        result_per_id = {str(k): FitResult(**v) for k, v in raw.items()}
         return result_per_id  # Dict[str, SimpleNamespace]
 
     def forecast_model_ind(

{datfid-0.1.16.dist-info → datfid-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datfid
-Version: 0.1.16
+Version: 0.1.18
 Summary: SDK to access the DATFID API hosted on Hugging Face Spaces
 Author: DATFID
 Author-email: igor.schapiro@datfid.com

datfid-0.1.18.dist-info/RECORD

@@ -0,0 +1,7 @@
+datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
+datfid/client.py,sha256=J-ite930zdIlU-ExEhcWCdhIIITxqyXuUAIdDKeCpoI,9477
+datfid-0.1.18.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
+datfid-0.1.18.dist-info/METADATA,sha256=qU93KNxZgZNSS2Fq2txjRJob3OyKNTTowioPqTkN36A,5146
+datfid-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datfid-0.1.18.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
+datfid-0.1.18.dist-info/RECORD,,

datfid-0.1.16.dist-info/RECORD

@@ -1,7 +0,0 @@
-datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
-datfid/client.py,sha256=uza0OZBtm5aQF0xdFF2TIBUOYeQ9zYmX-hS8-OBXWqw,8267
-datfid-0.1.16.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
-datfid-0.1.16.dist-info/METADATA,sha256=frz6VkxxFwrNeY7-ZbN2U4H7nCwN3-DIH9WkEQyHJtU,5146
-datfid-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datfid-0.1.16.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
-datfid-0.1.16.dist-info/RECORD,,