datfid 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

datfid/client.py

@@ -144,3 +144,105 @@ class DATFIDClient:
         except Exception as e:
             self.logger.error(f"Forecast generation failed: {str(e)}")
             raise
+
+    def fit_model_ind(
+        self,
+        df: pd.DataFrame,
+        id_col: str,
+        time_col: str,
+        y: str,
+        lag_y: Optional[Union[int, str, list[int]]] = None,
+        lagged_features: Optional[Dict[str, int]] = None,
+        current_features: Optional[list] = None,
+        filter_by_significance: bool = False,
+        meanvar_test: bool = False
+    ) -> SimpleNamespace:
+        """
+        Fit a model individual by individual using the DATFID API.
+        
+        Args:
+            df: DataFrame containing the data
+            id_col: Name of the ID column
+            time_col: Name of the time column
+            y: Name of the target variable
+            lagged_features: Dictionary of features and their lag values
+            current_features: List of current features to use
+            filter_by_significance: Whether to filter features by significance
+            meanvar_test: Whether to perform mean-variance test
+            
+        Returns:
+            SimpleNamespace containing the model fit results
+        """
+
+        df = df.copy()
+        for col in df.columns:
+            if pd.api.types.is_datetime64_any_dtype(df[col]):
+                df[col] = df[col].astype(str)
+
+        data = {
+            "df": df.to_dict(orient="records"),
+            "id_col": id_col,
+            "time_col": time_col,
+            "y": y,
+            "lag_y": lag_y,
+            "lagged_features": lagged_features or {},
+            "current_features": current_features or [],
+            "filter_by_significance": filter_by_significance,
+            "meanvar_test": meanvar_test
+        }
+        
+        response = requests.post(
+            f"{self.api_url}modelfit_ind/",
+            json=data,
+            headers=self.headers
+        )
+        
+        if response.status_code != 200:
+            raise Exception(f"Model fit failed: {response.text}")
+            
+        result_dict = response.json()
+        return SimpleNamespace(**result_dict)
+
+    def forecast_model_ind(
+        self,
+        df_forecast: pd.DataFrame
+    ) -> pd.DataFrame:
+        """
+        Generate forecasts using the fitted individual by individual model.
+        
+        Args:
+            df_forecast: DataFrame containing the forecast data
+            
+        Returns:
+            DataFrame containing the forecast results
+        """
+
+        try:
+            df_forecast = df_forecast.copy()
+            for col in df_forecast.columns:
+                if pd.api.types.is_datetime64_any_dtype(df_forecast[col]):
+                    df_forecast[col] = df_forecast[col].astype(str)
+
+            # Convert DataFrame to list of records
+            data = df_forecast.to_dict(orient="records")
+            
+            response = requests.post(
+                f"{self.api_url}modelforecast_ind/",
+                json=data,
+                headers=self.headers
+            )
+            
+            if response.status_code != 200:
+                raise Exception(f"Forecast generation failed: {response.text}")
+            
+            result = pd.DataFrame(response.json())
+            
+            # Clean up memory after operation
+            del df_forecast
+            del data
+            self._cleanup_memory()
+            
+            return result
+        except Exception as e:
+            self.logger.error(f"Forecast generation failed: {str(e)}")
+            raise

{datfid-0.1.13.dist-info → datfid-0.1.15.dist-info}/METADATA

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

datfid-0.1.15.dist-info/RECORD

@@ -0,0 +1,7 @@
+datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
+datfid/client.py,sha256=TG5uQ7CdGPPzT6AjcDIEutif1Su6b7SIlWz1ESyKeRY,8053
+datfid-0.1.15.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
+datfid-0.1.15.dist-info/METADATA,sha256=6Jp03iucoUMM7ewaNhOce2ju9G8zjfErSFqZj18JlHU,5146
+datfid-0.1.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datfid-0.1.15.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
+datfid-0.1.15.dist-info/RECORD,,

datfid-0.1.13.dist-info/RECORD

@@ -1,7 +0,0 @@
-datfid/__init__.py,sha256=PojdUQsimZvawR8lgqVpzs3EkRtkA7iwuDk7fovJF54,34
-datfid/client.py,sha256=XH8wTBsHOHT3Fl1DzDWP9PfIUxJGg24eLH7_cWuymic,4656
-datfid-0.1.13.dist-info/licenses/LICENSE,sha256=5b2JLb1-P8Nt919phjUmP0umS82SxGMlIF2QFn9ILgM,1084
-datfid-0.1.13.dist-info/METADATA,sha256=fleRinV54vP271uLNgLk5sTyQDy4zxw-edkcF3dSIjE,5146
-datfid-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datfid-0.1.13.dist-info/top_level.txt,sha256=VFG23u9NMNQJ8S_IFfW-S5d8oSUd71SFA346ovh_uxA,7
-datfid-0.1.13.dist-info/RECORD,,