Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-17 16:35:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update hyperparams info

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-04-11 19:35:39 +02:00
parent d04cb389c0
commit 947d54202d
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/fimdlp/mdlp.py
View File

@@ -115,6 +115,8 @@ 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):
@@ -251,11 +253,13 @@ 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()