scikit-survival 0.23.1__cp313-cp313-macosx_11_0_arm64.whl

sksurv/preprocessing.py

@@ -0,0 +1,155 @@
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+from sklearn.base import BaseEstimator, TransformerMixin
+from sklearn.utils.validation import _check_feature_names_in, check_is_fitted
+
+from .column import encode_categorical
+
+__all__ = ["OneHotEncoder"]
+
+
+def check_columns_exist(actual, expected):
+    missing_features = expected.difference(actual)
+    if len(missing_features) != 0:
+        raise ValueError(f"{len(missing_features)} features are missing from data: {missing_features.tolist()}")
+
+
+class OneHotEncoder(BaseEstimator, TransformerMixin):
+    """Encode categorical columns with `M` categories into `M-1` columns according
+    to the one-hot scheme.
+
+    The order of non-categorical columns is preserved, encoded columns are inserted
+    inplace of the original column.
+
+    Parameters
+    ----------
+    allow_drop : boolean, optional, default: True
+        Whether to allow dropping categorical columns that only consist
+        of a single category.
+
+    Attributes
+    ----------
+    feature_names_ : pandas.Index
+        List of encoded columns.
+
+    categories_ : dict
+        Categories of encoded columns.
+
+    encoded_columns_ : list
+        Name of columns after encoding.
+        Includes names of non-categorical columns.
+
+    n_features_in_ : int
+        Number of features seen during ``fit``.
+
+    feature_names_in_ : ndarray of shape (`n_features_in_`,)
+        Names of features seen during ``fit``. Defined only when `X`
+        has feature names that are all strings.
+    """
+
+    def __init__(self, *, allow_drop=True):
+        self.allow_drop = allow_drop
+
+    def fit(self, X, y=None):  # pylint: disable=unused-argument
+        """Retrieve categorical columns.
+
+        Parameters
+        ----------
+        X : pandas.DataFrame
+            Data to encode.
+        y :
+            Ignored. For compatibility with Pipeline.
+        Returns
+        -------
+        self : object
+            Returns self
+        """
+        self.fit_transform(X)
+        return self
+
+    def _encode(self, X, columns_to_encode):
+        return encode_categorical(X, columns=columns_to_encode, allow_drop=self.allow_drop)
+
+    def fit_transform(self, X, y=None, **fit_params):  # pylint: disable=unused-argument
+        """Convert categorical columns to numeric values.
+
+        Parameters
+        ----------
+        X : pandas.DataFrame
+            Data to encode.
+        y :
+            Ignored. For compatibility with TransformerMixin.
+        fit_params :
+            Ignored. For compatibility with TransformerMixin.
+
+        Returns
+        -------
+        Xt : pandas.DataFrame
+            Encoded data.
+        """
+        self._check_feature_names(X, reset=True)
+        self._check_n_features(X, reset=True)
+        columns_to_encode = X.select_dtypes(include=["object", "category"]).columns
+        x_dummy = self._encode(X, columns_to_encode)
+
+        self.feature_names_ = columns_to_encode
+        self.categories_ = {k: X[k].cat.categories for k in columns_to_encode}
+        self.encoded_columns_ = x_dummy.columns
+        return x_dummy
+
+    def transform(self, X):
+        """Convert categorical columns to numeric values.
+
+        Parameters
+        ----------
+        X : pandas.DataFrame
+            Data to encode.
+
+        Returns
+        -------
+        Xt : pandas.DataFrame
+            Encoded data.
+        """
+        check_is_fitted(self, "encoded_columns_")
+        self._check_n_features(X, reset=False)
+        check_columns_exist(X.columns, self.feature_names_)
+
+        Xt = X.copy()
+        for col, cat in self.categories_.items():
+            Xt[col] = Xt[col].cat.set_categories(cat)
+
+        new_data = self._encode(Xt, self.feature_names_)
+        return new_data.loc[:, self.encoded_columns_]
+
+    def get_feature_names_out(self, input_features=None):
+        """Get output feature names for transformation.
+
+        Parameters
+        ----------
+        input_features : array-like of str or None, default=None
+            Input features.
+
+            - If `input_features` is `None`, then `feature_names_in_` is
+              used as feature names in.
+            - If `input_features` is an array-like, then `input_features` must
+              match `feature_names_in_` if `feature_names_in_` is defined.
+
+        Returns
+        -------
+        feature_names_out : ndarray of str objects
+            Transformed feature names.
+        """
+        check_is_fitted(self, "encoded_columns_")
+        input_features = _check_feature_names_in(self, input_features)
+
+        return self.encoded_columns_.values.copy()

sksurv/svm/__init__.py

@@ -0,0 +1,11 @@
+from .minlip import HingeLossSurvivalSVM, MinlipSurvivalAnalysis
+from .naive_survival_svm import NaiveSurvivalSVM
+from .survival_svm import FastKernelSurvivalSVM, FastSurvivalSVM
+
+__all__ = [
+    "FastKernelSurvivalSVM",
+    "FastSurvivalSVM",
+    "HingeLossSurvivalSVM",
+    "MinlipSurvivalAnalysis",
+    "NaiveSurvivalSVM",
+]