upgini 1.2.111__py3-none-any.whl → 1.2.113__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.111"
+__version__ = "1.2.113"

upgini/autofe/date.py

@@ -244,7 +244,8 @@ class DateListDiffBounded(DateListDiff, ParametrizedOperator):
 
 class DatePercentileBase(PandasOperator, abc.ABC):
     is_binary: bool = True
-    output_type: Optional[str] = "float"
+    is_categorical: bool = True
+    output_type: Optional[str] = "category"
 
     date_unit: Optional[str] = None
 
@@ -254,7 +255,12 @@ class DatePercentileBase(PandasOperator, abc.ABC):
 
         bounds = self._get_bounds(left)
 
-        return right.index.to_series().apply(lambda i: self._perc(right[i], bounds[i]))
+        return (
+            right.index.to_series()
+            .apply(lambda i: self._perc(right[i], bounds[i]))
+            .astype(pd.Int64Dtype())
+            .astype("category")
+        )
 
     @abc.abstractmethod
     def _get_bounds(self, date_col: pd.Series) -> pd.Series:

upgini/dataset.py

@@ -50,7 +50,7 @@ except Exception:
 
 class Dataset:
     MIN_ROWS_COUNT = 100
-    MAX_ROWS = 200_000
+    MAX_ROWS = 100_000
     IMBALANCE_THESHOLD = 0.6
     MIN_TARGET_CLASS_ROWS = 100
     MAX_MULTICLASS_CLASS_COUNT = 100

upgini/utils/sample_utils.py

@@ -22,10 +22,10 @@ MULTICLASS_MIN_SAMPLE_THRESHOLD = 25_000
 BINARY_BOOTSTRAP_LOOPS = 5
 MULTICLASS_BOOTSTRAP_LOOPS = 2
 
-FIT_SAMPLE_THRESHOLD = 200_000
-FIT_SAMPLE_ROWS = 200_000
-FIT_SAMPLE_ROWS_WITH_EVAL_SET = 200_000
-FIT_SAMPLE_THRESHOLD_WITH_EVAL_SET = 200_000
+FIT_SAMPLE_THRESHOLD = 100_000
+FIT_SAMPLE_ROWS = 100_000
+FIT_SAMPLE_ROWS_WITH_EVAL_SET = 100_000
+FIT_SAMPLE_THRESHOLD_WITH_EVAL_SET = 100_000
 
 
 @dataclass

{upgini-1.2.111.dist-info → upgini-1.2.113.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.111
+Version: 1.2.113
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/
@@ -874,7 +874,7 @@ Some convenient ways to start contributing are:
 ⚙️ **Gitpod** [![Gitpod Ready-to-Code](https://img.shields.io/badge/Gitpod-Ready--to--Code-blue?logo=gitpod)](https://gitpod.io/#https://github.com/upgini/upgini) You can use Gitpod to launch a fully functional development environment right in your browser.
 
 ## 🔗 Useful links
-- [Simple sales predictions as a template notebook](#-simple-sales-predictions-use-as-a-template)
+- [Simple sales predictions as a template notebook](#-simple-sales-prediction-for-retail-stores)
 - [Full list of Kaggle Guides & Examples](https://www.kaggle.com/romaupgini/code)
 - [Project on PyPI](https://pypi.org/project/upgini)
 - [More perks for registered users](https://profile.upgini.com)

{upgini-1.2.111.dist-info → upgini-1.2.113.dist-info}/RECORD

@@ -1,7 +1,7 @@
-upgini/__about__.py,sha256=4KgOTmI-DrbwzTYsi8NTo4T_nHot9MIavj4EH29sgYA,24
+upgini/__about__.py,sha256=kaFCe2X7oNqZHww74j5Dd4YV4nG48zI57948z7H1dl4,24
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
-upgini/dataset.py,sha256=e6JDYTZ2AwC5aF-dqclKZKkiKrHo2f6cFmMQO2ZZmjM,32724
+upgini/dataset.py,sha256=xFi0a-A3uvtxVwFM6JOyitkEPd1I2slIBj5SWfys3hQ,32724
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
 upgini/features_enricher.py,sha256=rfVdHgUYEq9saqhWcI04jUmNQcAAn5Kto4w3WpxlOpA,221762
 upgini/http.py,sha256=zeAZvT6IAzOs9jQ3WG8mJBANLajgvv2LZePFzKz004w,45482
@@ -15,7 +15,7 @@ upgini/ads_management/ads_manager.py,sha256=igVbN2jz80Umb2BUJixmJVj-zx8unoKpecVo
 upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operators.py,sha256=rdjF5eaE4bC6Q4eu_el5Z7ekYt8DjOFermz2bePPbUc,333
 upgini/autofe/binary.py,sha256=oOEECc4nRzZN2tYaiqx8F2XHnfWpk1bVvb7ZkZJ0lO8,7709
-upgini/autofe/date.py,sha256=Ga022BUSgXJ4W3P8uWkPNo6k6J0IuEZw6Ezs9KNikPk,11188
+upgini/autofe/date.py,sha256=RvexgrL1_6ISYPVrl9HUQmPgpVSGQsTNv8YhNQWs-5M,11329
 upgini/autofe/feature.py,sha256=b4Ps_sCPui9b4h0K3ya85cfL1SWpLVrlHc40zkKVfAY,16329
 upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
 upgini/autofe/operator.py,sha256=RB3rKMjFi5Cx81RiYXN3OTCuXjmvzmFKQrxn4h0Oclo,5219
@@ -64,14 +64,14 @@ upgini/utils/mstats.py,sha256=u3gQVUtDRbyrOQK6V1UJ2Rx1QbkSNYGjXa6m3Z_dPVs,6286
 upgini/utils/phone_utils.py,sha256=IrbztLuOJBiePqqxllfABWfYlfAjYevPhXKipl95wUI,10432
 upgini/utils/postal_code_utils.py,sha256=5M0sUqH2DAr33kARWCTXR-ACyzWbjDq_-0mmEml6ZcU,1716
 upgini/utils/progress_bar.py,sha256=N-Sfdah2Hg8lXP_fV9EfUTXz_PyRt4lo9fAHoUDOoLc,1550
-upgini/utils/sample_utils.py,sha256=ETLPKQU_YngiYbdlnEoF2h7QS-3oN8et54q3Qs2ZAbA,15417
+upgini/utils/sample_utils.py,sha256=lZJ4yf9Jiq9Em2Ny9m3RIiF7WSxBPrc4E3xxn_8sQk8,15417
 upgini/utils/sklearn_ext.py,sha256=jLJWAKkqQinV15Z4y1ZnsN3c-fKFwXTsprs00COnyVU,49315
 upgini/utils/sort.py,sha256=8uuHs2nfSMVnz8GgvbOmgMB1PgEIZP1uhmeRFxcwnYw,7039
 upgini/utils/target_utils.py,sha256=i3Xt5l9ybB2_nF_ma5cfPuL3OeFTs2dY2xDI0p4Azpg,9049
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
 upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
-upgini-1.2.111.dist-info/METADATA,sha256=FFU_9hZGBiuvJFZpxnlMJVkWlZ4ykBOnA0KcGqOBSyQ,49529
-upgini-1.2.111.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.111.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.111.dist-info/RECORD,,
+upgini-1.2.113.dist-info/METADATA,sha256=p2gvqfKWenNyt1s95-ZalIp6mYCaq0LSVrzphZAQvH4,49528
+upgini-1.2.113.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+upgini-1.2.113.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.113.dist-info/RECORD,,

{upgini-1.2.111.dist-info → upgini-1.2.113.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.25.0
+Generator: hatchling 1.24.2
 Root-Is-Purelib: true
 Tag: py3-none-any