Update hyperparams info

This commit is contained in:
2023-04-11 19:35:39 +02:00
parent d04cb389c0
commit 947d54202d

View File

@@ -27,7 +27,7 @@ class FImdlp(TransformerMixin, BaseEstimator):
max_depth: int, default=1e6
The maximum depth of the discretization process.
max_cuts: float, default=0
The maximum number of cut points to be computed for each feature.
The maximum number of cut points to be computed for each feature.
Attributes
----------
@@ -115,6 +115,8 @@ class FImdlp(TransformerMixin, BaseEstimator):
)
# target of every feature. Start with -1 => y (see join_fit)
self.target_ = [-1] * self.n_features_in_
# target of every feature. Start with -1 => y (see join_fit)
self.target_ = [-1] * self.n_features_in_
return self
def _fit_discretizer(self, feature):
@@ -251,11 +253,13 @@ class FImdlp(TransformerMixin, BaseEstimator):
)
if target in features:
raise ValueError("Target cannot be in features to join")
raise ValueError("Target cannot be in features to join")
y_join = [
f"{str(item_y)}{''.join([str(x) for x in items_x])}".encode()
for item_y, items_x in zip(self.y_, data[:, features])
]
self.target_[target] = features + [-1]
self.target_[target] = features + [-1]
self.y_join_ = y_join
self.discretizer_[target].fit(self.X_[:, target], factorize(y_join))
self.cut_points_[target] = self.discretizer_[target].get_cut_points()