wavetrainer 0.0.7__tar.gz → 0.0.9__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (57) hide show
  1. {wavetrainer-0.0.7/wavetrainer.egg-info → wavetrainer-0.0.9}/PKG-INFO +1 -1
  2. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/setup.py +1 -1
  3. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/__init__.py +1 -1
  4. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/catboost_model.py +8 -1
  5. {wavetrainer-0.0.7 → wavetrainer-0.0.9/wavetrainer.egg-info}/PKG-INFO +1 -1
  6. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/LICENSE +0 -0
  7. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/MANIFEST.in +0 -0
  8. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/README.md +0 -0
  9. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/requirements.txt +0 -0
  10. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/setup.cfg +0 -0
  11. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/tests/__init__.py +0 -0
  12. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/tests/model/__init__.py +0 -0
  13. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/tests/model/catboost_kwargs_test.py +0 -0
  14. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/tests/trainer_test.py +0 -0
  15. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/calibrator/__init__.py +0 -0
  16. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/calibrator/calibrator.py +0 -0
  17. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/calibrator/calibrator_router.py +0 -0
  18. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/calibrator/mapie_calibrator.py +0 -0
  19. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/calibrator/vennabers_calibrator.py +0 -0
  20. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/create.py +0 -0
  21. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/exceptions.py +0 -0
  22. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/fit.py +0 -0
  23. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/__init__.py +0 -0
  24. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/catboost_classifier_wrap.py +0 -0
  25. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/catboost_kwargs.py +0 -0
  26. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/catboost_regressor_wrap.py +0 -0
  27. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/model.py +0 -0
  28. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/model_router.py +0 -0
  29. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model_type.py +0 -0
  30. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/params.py +0 -0
  31. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/__init__.py +0 -0
  32. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/base_selector_reducer.py +0 -0
  33. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/combined_reducer.py +0 -0
  34. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/constant_reducer.py +0 -0
  35. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/correlation_reducer.py +0 -0
  36. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/duplicate_reducer.py +0 -0
  37. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/nonnumeric_reducer.py +0 -0
  38. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/reducer/reducer.py +0 -0
  39. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/selector/__init__.py +0 -0
  40. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/selector/selector.py +0 -0
  41. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/trainer.py +0 -0
  42. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/__init__.py +0 -0
  43. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/class_weights.py +0 -0
  44. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/combined_weights.py +0 -0
  45. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/exponential_weights.py +0 -0
  46. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/linear_weights.py +0 -0
  47. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/noop_weights.py +0 -0
  48. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/sigmoid_weights.py +0 -0
  49. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/weights.py +0 -0
  50. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/weights/weights_router.py +0 -0
  51. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/windower/__init__.py +0 -0
  52. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/windower/windower.py +0 -0
  53. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer.egg-info/SOURCES.txt +0 -0
  54. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer.egg-info/dependency_links.txt +0 -0
  55. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer.egg-info/not-zip-safe +0 -0
  56. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer.egg-info/requires.txt +0 -0
  57. {wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer.egg-info/top_level.txt +0 -0
{wavetrainer-0.0.7/wavetrainer.egg-info → wavetrainer-0.0.9}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: wavetrainer
3
- Version: 0.0.7
3
+ Version: 0.0.9
4
4
  Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
5
5
  Home-page: https://github.com/8W9aG/wavetrainer
6
6
  Author: Will Sackfield
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/setup.py RENAMED
@@ -23,7 +23,7 @@ def install_requires() -> typing.List[str]:
23
23
 
24
24
  setup(
25
25
  name='wavetrainer',
26
- version='0.0.7',
26
+ version='0.0.9',
27
27
  description='A library for automatically finding the optimal model within feature and hyperparameter space.',
28
28
  long_description=long_description,
29
29
  long_description_content_type='text/markdown',
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/__init__.py RENAMED
@@ -2,5 +2,5 @@
2
2
 
3
3
  from .create import create
4
4
 
5
- __VERSION__ = "0.0.7"
5
+ __VERSION__ = "0.0.9"
6
6
  __all__ = ("create",)
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/wavetrainer/model/catboost_model.py RENAMED
@@ -123,6 +123,8 @@ class CatboostModel(Model):
123
123
  ) -> Self:
124
124
  if y is None:
125
125
  raise ValueError("y is null.")
126
+ if eval_x is None:
127
+ raise ValueError("eval_x is null.")
126
128
  self._model_type = determine_model_type(y)
127
129
  catboost = self._provide_catboost()
128
130
 
@@ -130,10 +132,12 @@ class CatboostModel(Model):
130
132
  df,
131
133
  label=y,
132
134
  weight=w,
135
+ cat_features=df.select_dtypes(include="category").columns.tolist(),
133
136
  )
134
137
  eval_pool = Pool(
135
138
  eval_x,
136
139
  label=eval_y,
140
+ cat_features=eval_x.select_dtypes(include="category").columns.tolist(),
137
141
  )
138
142
  catboost.fit(
139
143
  train_pool,
@@ -145,7 +149,10 @@ class CatboostModel(Model):
145
149
  return self
146
150
 
147
151
  def transform(self, df: pd.DataFrame) -> pd.DataFrame:
148
- pred_pool = Pool(df)
152
+ pred_pool = Pool(
153
+ df,
154
+ cat_features=df.select_dtypes(include="category").columns.tolist(),
155
+ )
149
156
  catboost = self._provide_catboost()
150
157
  pred = catboost.predict(pred_pool)
151
158
  df = pd.DataFrame(
{wavetrainer-0.0.7 → wavetrainer-0.0.9/wavetrainer.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: wavetrainer
3
- Version: 0.0.7
3
+ Version: 0.0.9
4
4
  Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
5
5
  Home-page: https://github.com/8W9aG/wavetrainer
6
6
  Author: Will Sackfield
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/LICENSE RENAMED
File without changes
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/MANIFEST.in RENAMED
File without changes
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/README.md RENAMED
File without changes
{wavetrainer-0.0.7 → wavetrainer-0.0.9}/setup.cfg RENAMED
File without changes