datfid 0.1.13__tar.gz → 0.1.15__tar.gz

{datfid-0.1.13 → datfid-0.1.15}/PKG-INFO

@@ -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.13 → datfid-0.1.15}/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 → datfid-0.1.15}/datfid.egg-info/PKG-INFO

@@ -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.13 → datfid-0.1.15}/setup.py

@@ -2,11 +2,11 @@ from setuptools import setup, find_packages
 
 # to build:
 # 1) open this file at level of datfid-sdk folder
-# 2) change version in this file
+# 2) change version in this file and save it
 # 3) delete folder datfid.egg-info
 # 4) delete older files from dist folder
 # 5) in terminal: python setup.py sdist bdist_wheel
-# 6) in terminal: twine upload --repository testpypi dist/*
+# 6) in terminal: twine upload --repository pypi dist/*
 # 7) in hugging face delete older files from dist folder
 # 8) in hugging face upload updated files
 # 9) in terminal uninstall older version of datfid: pip uninstall datfid
@@ -17,7 +17,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="datfid",
-    version="0.1.13",
+    version="0.1.15",
     description="SDK to access the DATFID API hosted on Hugging Face Spaces",
     long_description=long_description,
     long_description_content_type="text/markdown",  # Important!