InsurAutoML 0.2.6__tar.gz → 0.2.8__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 (106) hide show
  1. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/__init__.py +9 -4
  2. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/base.py +64 -63
  3. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/constant.py +11 -5
  4. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/datasets/tabular.py +289 -829
  5. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/encoding/__init__.py +9 -5
  6. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/encoding/base.py +20 -4
  7. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/encoding/encoding.py +12 -15
  8. insurautoml-0.2.8/InsurAutoML/ext/__init__.py +42 -0
  9. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/__init__.py +41 -0
  10. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_encoder_hyperparameter.py → insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit/__init__.py +9 -9
  11. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit/split.py +154 -0
  12. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit/utils.py +105 -0
  13. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit_cpp/bindings.cpp +146 -0
  14. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit_cpp/sPlit.cpp +140 -0
  15. insurautoml-0.2.8/InsurAutoML/ext/suppsplit/suppsplit_cpp/sp.cpp +246 -0
  16. insurautoml-0.2.8/InsurAutoML/ext/twinreduction/__init__.py +41 -0
  17. insurautoml-0.2.8/InsurAutoML/ext/twinreduction/twinning_cpp/bindings.cpp +101 -0
  18. insurautoml-0.2.8/InsurAutoML/ext/twinreduction/twinning_cpp/twinning.cpp +163 -0
  19. insurautoml-0.2.8/InsurAutoML/ext/twinreduction/twinreduction/__init__.py +42 -0
  20. insurautoml-0.2.8/InsurAutoML/ext/twinreduction/twinreduction/twinreduction.py +131 -0
  21. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/__init__.py +17 -14
  22. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/filter.py +181 -39
  23. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/wrapper.py +2 -5
  24. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/ML.py +18 -17
  25. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/__init__.py +11 -4
  26. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/base.py +68 -25
  27. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/ensemble.py +3 -9
  28. insurautoml-0.2.8/InsurAutoML/hpo/informed/ML.py +403 -0
  29. insurautoml-0.2.8/InsurAutoML/hpo/informed/__init__.py +45 -0
  30. insurautoml-0.2.8/InsurAutoML/hpo/informed/base.py +1445 -0
  31. insurautoml-0.2.8/InsurAutoML/hpo/informed/fixed.py +177 -0
  32. insurautoml-0.2.8/InsurAutoML/hpo/informed/informed_ensemble.py +472 -0
  33. insurautoml-0.2.8/InsurAutoML/hpo/informed/utils.py +472 -0
  34. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/utils.py +5 -4
  35. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hyperparameters/__init__.py +9 -3
  36. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/imputation/__init__.py +6 -5
  37. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/imputation/base.py +37 -12
  38. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/imputation/multiple.py +232 -138
  39. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/FNN.py +11 -11
  40. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/RNN.py +11 -11
  41. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/__init__.py +14 -3
  42. insurautoml-0.2.8/InsurAutoML/model/catboost.py +112 -0
  43. insurautoml-0.2.8/InsurAutoML/prep/__init__.py +41 -0
  44. insurautoml-0.2.8/InsurAutoML/prep/pipeline.py +325 -0
  45. insurautoml-0.2.8/InsurAutoML/prep/prep.py +41 -0
  46. insurautoml-0.2.8/InsurAutoML/prep/utils.py +162 -0
  47. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/__init__.py +3 -2
  48. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/base.py +4 -5
  49. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/data.py +4 -5
  50. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/metadata.py +29 -25
  51. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/optimize.py +669 -17
  52. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/stats.py +308 -64
  53. {InsurAutoML-0.2.6 → insurautoml-0.2.8/InsurAutoML.egg-info}/PKG-INFO +144 -31
  54. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML.egg-info/SOURCES.txt +26 -9
  55. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML.egg-info/requires.txt +13 -2
  56. InsurAutoML-0.2.6/README.md → insurautoml-0.2.8/PKG-INFO +342 -208
  57. InsurAutoML-0.2.6/InsurAutoML.egg-info/PKG-INFO → insurautoml-0.2.8/README.md +259 -229
  58. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/setup.py +138 -18
  59. insurautoml-0.2.8/tests/test_base.py +72 -0
  60. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/__init__.py +0 -47
  61. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_balancing_hyperparameter.py +0 -105
  62. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_classifier_hyperparameter.py +0 -269
  63. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_feature_selection_hyperparameter.py +0 -272
  64. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_imputer_hyperparameter.py +0 -74
  65. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_regressor_hyperparameter.py +0 -282
  66. InsurAutoML-0.2.6/InsurAutoML/hyperparameters/_hyperopt/_scaling_hyperparameter.py +0 -55
  67. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/balancing/__init__.py +0 -0
  68. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/balancing/base.py +0 -0
  69. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/balancing/mixed_sampling.py +0 -0
  70. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/balancing/over_sampling.py +0 -0
  71. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/balancing/under_sampling.py +0 -0
  72. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/datasets/__init__.py +0 -0
  73. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/datasets/image.py +0 -0
  74. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/datasets/text.py +0 -0
  75. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/autosklearn.py +0 -0
  76. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/base.py +0 -0
  77. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/embed.py +0 -0
  78. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/hybrid.py +0 -0
  79. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/feature_selection/sklearn.py +0 -0
  80. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/hpo/DL.py +0 -0
  81. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/imputation/clustering.py +0 -0
  82. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/imputation/nn.py +0 -0
  83. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/ML.py +0 -0
  84. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/autosklearn.py +0 -0
  85. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/base.py +0 -0
  86. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/clustering.py +0 -0
  87. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/gam.py +0 -0
  88. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/lightgbm.py +0 -0
  89. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/sklearn_classifiers.py +0 -0
  90. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/sklearn_regressors.py +0 -0
  91. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/model/xgboost.py +0 -0
  92. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/scaling/__init__.py +0 -0
  93. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/scaling/base.py +0 -0
  94. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/scaling/scaling.py +0 -0
  95. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/_c/__init__.py +0 -0
  96. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/_c/_base.pyx +0 -0
  97. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/_c/_stat.pyx +0 -0
  98. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/eda.py +0 -0
  99. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/file.py +0 -0
  100. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/utils/tensor.py +0 -0
  101. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML/version.py +0 -0
  102. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML.egg-info/dependency_links.txt +0 -0
  103. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML.egg-info/not-zip-safe +0 -0
  104. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/InsurAutoML.egg-info/top_level.txt +0 -0
  105. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/LICENSE +0 -0
  106. {InsurAutoML-0.2.6 → insurautoml-0.2.8}/setup.cfg +0 -0
@@ -5,19 +5,19 @@ GitHub: https://github.com/PanyiDong/
5
5
  Mathematics Department, University of Illinois at Urbana-Champaign (UIUC)
6
6
 
7
7
  Project: InsurAutoML
8
- Latest Version: 0.2.5
8
+ Latest Version: 0.2.6
9
9
  Relative Path: /InsurAutoML/__init__.py
10
10
  File Created: Wednesday, 16th November 2022 7:23:10 pm
11
11
  Author: Panyi Dong (panyid2@illinois.edu)
12
12
 
13
13
  -----
14
- Last Modified: Thursday, 25th May 2023 11:04:19 am
14
+ Last Modified: Thursday, 4th December 2025 1:14:46 pm
15
15
  Modified By: Panyi Dong (panyid2@illinois.edu)
16
16
 
17
17
  -----
18
18
  MIT License
19
19
 
20
- Copyright (c) 2022 - 2022, Panyi Dong
20
+ Copyright (c) 2022 - 2025, Panyi Dong
21
21
 
22
22
  Permission is hereby granted, free of charge, to any person obtaining a copy
23
23
  of this software and associated documentation files (the "Software"), to deal
@@ -38,13 +38,15 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
38
38
  SOFTWARE.
39
39
  """
40
40
 
41
-
42
41
  from .version import _get_version
43
42
  from .base import no_processing, load_data, set_seed
44
43
  from .hpo import (
45
44
  AutoTabular,
46
45
  AutoTabularClassifier,
47
46
  AutoTabularRegressor,
47
+ InformedAutoTabular,
48
+ InformedAutoTabularRegressor,
49
+ InformedAutoTabularClassifier,
48
50
  # AutoTextClassifier,
49
51
  # AutoNextWordPrediction,
50
52
  )
@@ -257,6 +259,9 @@ __all__ = [
257
259
  "AutoTabular", # _model_selection
258
260
  "AutoTabularClassifier",
259
261
  "AutoTabularRegressor",
262
+ "InformedAutoTabular",
263
+ "InformedAutoTabularRegressor",
264
+ "InformedAutoTabularClassifier",
260
265
  "AutoTextClassifier",
261
266
  "AutoNextWordPrediction",
262
267
  ]
@@ -5,13 +5,13 @@ GitHub: https://github.com/PanyiDong/
5
5
  Mathematics Department, University of Illinois at Urbana-Champaign (UIUC)
6
6
 
7
7
  Project: InsurAutoML
8
- Latest Version: 0.2.5
8
+ Latest Version: 0.2.6
9
9
  Relative Path: /InsurAutoML/base.py
10
10
  File Created: Monday, 24th October 2022 11:56:57 pm
11
11
  Author: Panyi Dong (panyid2@illinois.edu)
12
12
 
13
13
  -----
14
- Last Modified: Wednesday, 31st May 2023 1:49:15 pm
14
+ Last Modified: Wednesday, 12th November 2025 11:21:01 am
15
15
  Modified By: Panyi Dong (panyid2@illinois.edu)
16
16
 
17
17
  -----
@@ -49,14 +49,15 @@ import warnings
49
49
  # check if rpy2 available
50
50
  # if not, can not read R type data
51
51
  import importlib
52
-
53
- rpy2_spec = importlib.util.find_spec("rpy2")
54
- if rpy2_spec is not None:
55
- import rpy2
56
- import rpy2.robjects as ro
57
- from rpy2.robjects import Formula, pandas2ri
58
- from rpy2.robjects.conversion import localconverter
59
- from rpy2.robjects.packages import importr
52
+ # NOTE: Remove rda/rdata loading as rpy2 sometimes causes installation issues
53
+ # Uncomment the following lines if R data loading is needed
54
+ # rpy2_spec = importlib.util.find_spec("rpy2")
55
+ # if rpy2_spec is not None:
56
+ # import rpy2
57
+ # import rpy2.robjects as ro
58
+ # from rpy2.robjects import Formula, pandas2ri
59
+ # from rpy2.robjects.conversion import localconverter
60
+ # from rpy2.robjects.packages import importr
60
61
 
61
62
  torch_spec = importlib.util.find_spec("torch")
62
63
 
@@ -211,59 +212,59 @@ class load_data:
211
212
  _filename = os.path.split(_data_path)[-1]
212
213
  self.database[_filename.split(".")[0]] = pd.read_csv(_data_path)
213
214
 
214
- # load .rda/.rdata files in the path
215
- # will not read any files if rpy2 is not available
216
- if rpy2_spec is None:
217
- if self.data_type == ".rda" or self.data_type == ".rdata":
218
- raise ImportError("Require rpy2 package, package not found!")
219
- if self.data_type == "all":
220
- pass
221
- else:
222
- if (
223
- self.data_type == ".rda"
224
- or self.data_type == ".rdata"
225
- or self.data_type == "all"
226
- ):
227
- if self.data_type == ".rda" or self.data_type == "all":
228
- if filename is None:
229
- _rda_files = glob.glob(path + "*.rda")
230
- elif isinstance(filename, list):
231
- _rda_files = []
232
- for _filename in filename:
233
- _rda_files += glob.glob(path + _filename + ".rda")
234
- else:
235
- _rda_files = glob.glob(path + filename + ".rda")
236
- if self.data_type == ".rdata" or self.data_type == "all":
237
- if filename is None:
238
- _rdata_files = glob.glob(path + "*.rdata")
239
- elif isinstance(filename, list):
240
- _rdata_files = []
241
- for _filename in filename:
242
- _rdata_files += glob.glob(path + _filename + ".rdata")
243
- else:
244
- _rdata_files = glob.glob(path + filename + ".rdata")
245
-
246
- if not _rda_files and self.data_type == ".rda":
247
- warnings.warn("No .rda file found!")
248
- elif not _rdata_files and self.data_type == ".rdata":
249
- warnings.warn("No .rdata file found!")
250
- elif _rda_files + _rdata_files:
251
- for _data_path in _rda_files + _rdata_files:
252
- # in linux path, the path separator is '/'
253
- # in windows path, the path separator is '\\'
254
- # _filename = (
255
- # _data_path.split("/")[-1]
256
- # if "/" in _data_path
257
- # else _data_path.split("\\")[-1]
258
- # )
259
- # use os.path.split for unify path separator
260
- _filename = os.path.split(_data_path)[-1]
261
- ro.r('load("' + _data_path + '")')
262
- ro.r("rdata = " + _filename.split(".")[0])
263
- with localconverter(ro.default_converter + pandas2ri.converter):
264
- self.database[
265
- _filename.split(".")[0]
266
- ] = ro.conversion.rpy2py(ro.r.rdata)
215
+ # # load .rda/.rdata files in the path
216
+ # # will not read any files if rpy2 is not available
217
+ # if rpy2_spec is None:
218
+ # if self.data_type == ".rda" or self.data_type == ".rdata":
219
+ # raise ImportError("Require rpy2 package, package not found!")
220
+ # if self.data_type == "all":
221
+ # pass
222
+ # else:
223
+ # if (
224
+ # self.data_type == ".rda"
225
+ # or self.data_type == ".rdata"
226
+ # or self.data_type == "all"
227
+ # ):
228
+ # if self.data_type == ".rda" or self.data_type == "all":
229
+ # if filename is None:
230
+ # _rda_files = glob.glob(path + "*.rda")
231
+ # elif isinstance(filename, list):
232
+ # _rda_files = []
233
+ # for _filename in filename:
234
+ # _rda_files += glob.glob(path + _filename + ".rda")
235
+ # else:
236
+ # _rda_files = glob.glob(path + filename + ".rda")
237
+ # if self.data_type == ".rdata" or self.data_type == "all":
238
+ # if filename is None:
239
+ # _rdata_files = glob.glob(path + "*.rdata")
240
+ # elif isinstance(filename, list):
241
+ # _rdata_files = []
242
+ # for _filename in filename:
243
+ # _rdata_files += glob.glob(path + _filename + ".rdata")
244
+ # else:
245
+ # _rdata_files = glob.glob(path + filename + ".rdata")
246
+
247
+ # if not _rda_files and self.data_type == ".rda":
248
+ # warnings.warn("No .rda file found!")
249
+ # elif not _rdata_files and self.data_type == ".rdata":
250
+ # warnings.warn("No .rdata file found!")
251
+ # elif _rda_files + _rdata_files:
252
+ # for _data_path in _rda_files + _rdata_files:
253
+ # # in linux path, the path separator is '/'
254
+ # # in windows path, the path separator is '\\'
255
+ # # _filename = (
256
+ # # _data_path.split("/")[-1]
257
+ # # if "/" in _data_path
258
+ # # else _data_path.split("\\")[-1]
259
+ # # )
260
+ # # use os.path.split for unify path separator
261
+ # _filename = os.path.split(_data_path)[-1]
262
+ # ro.r('load("' + _data_path + '")')
263
+ # ro.r("rdata = " + _filename.split(".")[0])
264
+ # with localconverter(ro.default_converter + pandas2ri.converter):
265
+ # self.database[
266
+ # _filename.split(".")[0]
267
+ # ] = ro.conversion.rpy2py(ro.r.rdata)
267
268
 
268
269
  if self.data_type == "all" and not self.database:
269
270
  warnings.warn("No file found!")
@@ -5,19 +5,19 @@ GitHub: https://github.com/PanyiDong/
5
5
  Mathematics Department, University of Illinois at Urbana-Champaign (UIUC)
6
6
 
7
7
  Project: InsurAutoML
8
- Latest Version: 0.2.5
8
+ Latest Version: 0.2.6
9
9
  Relative Path: /InsurAutoML/constant.py
10
10
  File: _constant.py
11
11
  Author: Panyi Dong (panyid2@illinois.edu)
12
12
 
13
13
  -----
14
- Last Modified: Sunday, 10th December 2023 11:13:54 am
14
+ Last Modified: Tuesday, 16th December 2025 8:20:52 pm
15
15
  Modified By: Panyi Dong (panyid2@illinois.edu)
16
16
 
17
17
  -----
18
18
  MIT License
19
19
 
20
- Copyright (c) 2022 - 2022, Panyi Dong
20
+ Copyright (c) 2022 - 2025, Panyi Dong
21
21
 
22
22
  Permission is hereby granted, free of charge, to any person obtaining a copy
23
23
  of this software and associated documentation files (the "Software"), to deal
@@ -143,6 +143,9 @@ FEATURE_SELECTION = [
143
143
  "select_rates_classification",
144
144
  "select_rates_regression",
145
145
  "truncatedSVD",
146
+ "mRMR",
147
+ "CBFS",
148
+ "FOCI",
146
149
  ]
147
150
 
148
151
  # classifiers
@@ -205,7 +208,8 @@ PROP_UNIQUE = 0.1
205
208
 
206
209
  # maximum unique classes determined as categorical variable
207
210
  # 31 is capped by days in a month
208
- UNI_CLASS = 31
211
+ # 120 is capped by age of a person
212
+ UNI_CLASS = 120
209
213
 
210
214
  # maximum iteration allowed for the algorithm
211
215
  MAX_ITER = 1024
@@ -228,7 +232,7 @@ LIGHTGBM_REGRESSION = [
228
232
  "poisson",
229
233
  "quantile",
230
234
  "mape",
231
- "gamma",
235
+ # "gamma",
232
236
  "tweedie",
233
237
  ]
234
238
 
@@ -276,6 +280,8 @@ REGRESSION_CRITERIA = [
276
280
  # methods corresponding process must have
277
281
  METHOD_MAPPING = {
278
282
  "encoder": ["fit", "refit"],
283
+ "complete_prep": ["fit", "transform"],
284
+ "missing_prep": ["fit", "transform"],
279
285
  "imputer": ["fill"],
280
286
  "balancing": ["fit_transform"],
281
287
  "scaling": ["fit", "transform", "fit_transform"],