c50py 0.2.1__tar.gz → 0.2.2__tar.gz

{c50py-0.2.1/src/c50py.egg-info → c50py-0.2.2}/PKG-INFO

@@ -1,11 +1,11 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: c50py
-Version: 0.2.1
+Version: 0.2.2
 Summary: C5.0-like Decision Trees in Python (scikit-learn style)
 Author-email: David Díaz <daviddiazsolis@gmail.com>
 License: MIT
-Project-URL: Homepage, https://github.com/daviddiazsolis/c5py
-Project-URL: Issues, https://github.com/daviddiazsolis/c5py/issues
+Project-URL: Homepage, https://github.com/daviddiazsolis/c50py
+Project-URL: Issues, https://github.com/daviddiazsolis/c50py/issues
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
@@ -18,7 +18,6 @@ Requires-Dist: numpy>=1.21
 Requires-Dist: scikit-learn>=1.2
 Provides-Extra: graphviz
 Requires-Dist: graphviz>=0.20; extra == "graphviz"
-Dynamic: license-file
 
 # c5py — C5.0‑style Decision Trees for Python (clean 0.2.0)
 

{c50py-0.2.1 → c50py-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "c50py"
-version = "0.2.1"
+version = "0.2.2"
 description = "C5.0-like Decision Trees in Python (scikit-learn style)"
 readme = "README.md"
 license = { text = "MIT" }
@@ -27,8 +27,8 @@ classifiers = [
 graphviz = ["graphviz>=0.20"]
 
 [project.urls]
-Homepage = "https://github.com/daviddiazsolis/c5py"
-Issues = "https://github.com/daviddiazsolis/c5py/issues"
+Homepage = "https://github.com/daviddiazsolis/c50py"
+Issues = "https://github.com/daviddiazsolis/c50py/issues"
 
 [tool.setuptools.packages.find]
 where = ["src"]

{c50py-0.2.1 → c50py-0.2.2}/src/c50py/regressor.py

@@ -212,7 +212,7 @@ class C5Regressor:
 
     # ----------------------------- Public API -----------------------------
 
-    def fit(self, X, y, sample_weight: Optional[np.ndarray] = None):
+    def fit(self, X, y, sample_weight: Optional[np.ndarray] = None, feature_names: Optional[List[str]] = None):
         X = np.asarray(X, dtype=object)
         y = _as_float_array(y)
         n, m = X.shape
@@ -223,9 +223,16 @@ class C5Regressor:
             if w.shape[0] != n:
                 raise ValueError("sample_weight must have same length as y")
 
-        if self.feature_names is not None and len(self.feature_names) != m:
+        if feature_names is not None:
+            if len(feature_names) != m:
+                raise ValueError("feature_names length must match X.shape[1]")
+            self.feature_names_ = list(feature_names)
+        elif self.feature_names is not None and len(self.feature_names) != m:
             raise ValueError("feature_names length must match X.shape[1]")
-        self.feature_names_ = list(self.feature_names) if self.feature_names is not None else None
+        elif self.feature_names is not None:
+             self.feature_names_ = list(self.feature_names)
+        else:
+             self.feature_names_ = None
 
         self.is_cat_ = self._infer_categorical_features(X)
         # Gather categorical values up to cap

{c50py-0.2.1 → c50py-0.2.2}/src/c50py/tree.py

@@ -269,7 +269,7 @@ class C5Classifier(BaseEstimator, ClassifierMixin):
         self.classes_ = None
         self.n_features_ = None
 
-    def fit(self, X, y, sample_weight=None):
+    def fit(self, X, y, sample_weight=None, feature_names=None):
         X = np.asarray(X)
         y = np.asarray(y)
         if sample_weight is None:
@@ -281,7 +281,11 @@ class C5Classifier(BaseEstimator, ClassifierMixin):
 
         # Map categorical feature names to indices
         n_features = X.shape[1]
-        if self.feature_names_ is None:
+        if feature_names is not None:
+             if len(feature_names) != n_features:
+                 raise ValueError("feature_names length must match X.shape[1]")
+             self.feature_names_ = list(feature_names)
+        elif self.feature_names_ is None:
             self.feature_names_ = [f"f{i}" for i in range(n_features)]
         cf = self.categorical_features
         if cf is not None:

{c50py-0.2.1 → c50py-0.2.2/src/c50py.egg-info}/PKG-INFO

@@ -1,11 +1,11 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: c50py
-Version: 0.2.1
+Version: 0.2.2
 Summary: C5.0-like Decision Trees in Python (scikit-learn style)
 Author-email: David Díaz <daviddiazsolis@gmail.com>
 License: MIT
-Project-URL: Homepage, https://github.com/daviddiazsolis/c5py
-Project-URL: Issues, https://github.com/daviddiazsolis/c5py/issues
+Project-URL: Homepage, https://github.com/daviddiazsolis/c50py
+Project-URL: Issues, https://github.com/daviddiazsolis/c50py/issues
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
@@ -18,7 +18,6 @@ Requires-Dist: numpy>=1.21
 Requires-Dist: scikit-learn>=1.2
 Provides-Extra: graphviz
 Requires-Dist: graphviz>=0.20; extra == "graphviz"
-Dynamic: license-file
 
 # c5py — C5.0‑style Decision Trees for Python (clean 0.2.0)