Remove duplicated lines

This commit is contained in:
2023-04-11 19:45:37 +02:00
parent 947d54202d
commit fa8c4a221d

View File

@@ -115,8 +115,6 @@ class FImdlp(TransformerMixin, BaseEstimator):
) )
# target of every feature. Start with -1 => y (see join_fit) # target of every feature. Start with -1 => y (see join_fit)
self.target_ = [-1] * self.n_features_in_ 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 return self
def _fit_discretizer(self, feature): def _fit_discretizer(self, feature):
@@ -253,13 +251,11 @@ class FImdlp(TransformerMixin, BaseEstimator):
) )
if target in features: if target in features:
raise ValueError("Target cannot be in features to join") raise ValueError("Target cannot be in features to join")
raise ValueError("Target cannot be in features to join")
y_join = [ y_join = [
f"{str(item_y)}{''.join([str(x) for x in items_x])}".encode() f"{str(item_y)}{''.join([str(x) for x in items_x])}".encode()
for item_y, items_x in zip(self.y_, data[:, features]) for item_y, items_x in zip(self.y_, data[:, features])
] ]
self.target_[target] = features + [-1] self.target_[target] = features + [-1]
self.target_[target] = features + [-1]
self.y_join_ = y_join self.y_join_ = y_join
self.discretizer_[target].fit(self.X_[:, target], factorize(y_join)) self.discretizer_[target].fit(self.X_[:, target], factorize(y_join))
self.cut_points_[target] = self.discretizer_[target].get_cut_points() self.cut_points_[target] = self.discretizer_[target].get_cut_points()