From 6e1754856393742997050b403d3d38e1302bbe9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Tue, 25 Apr 2023 17:53:36 +0200
Subject: [PATCH] Add url to pyproject and comment to mdlp

---
 pyproject.toml     | 1 +
 src/fimdlp/mdlp.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 229bfd7..f677c57 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,6 +38,7 @@ classifiers = [
 
 [project.urls]
 Home = "https://github.com/doctorado-ml/FImdlp"
+Base = "https://github.com/rmontanana/mdlp"
 
 [tool.black]
 line-length = 79
diff --git a/src/fimdlp/mdlp.py b/src/fimdlp/mdlp.py
index 4e5503a..345a6f1 100644
--- a/src/fimdlp/mdlp.py
+++ b/src/fimdlp/mdlp.py
@@ -255,6 +255,7 @@ class FImdlp(TransformerMixin, BaseEstimator):
             f"{str(item_y)}{''.join([str(x) for x in items_x])}".encode()
             for item_y, items_x in zip(self.y_, data[:, features])
         ]
+        # Store in target_ the features used with class to discretize target
         self.target_[target] = features + [-1]
         self.y_join_ = y_join
         self.discretizer_[target].fit(self.X_[:, target], factorize(y_join))