multipers 2.3.3b6__cp312-cp312-manylinux_2_39_x86_64.whl

multipers/ml/accuracies.py

@@ -0,0 +1,90 @@
+import pandas as pd
+from warnings import warn
+import numpy as np
+from tqdm import tqdm
+from os.path import exists
+
+
+def accuracy_to_csv(
+    X,
+    Y,
+    cl,
+    k: float = 10,
+    dataset: str = "",
+    shuffle=True,
+    verbose: bool = True,
+    **more_columns,
+):
+    assert k > 0, "k is either the number of kfold > 1 or the test size > 0."
+    if k > 1:
+        k = int(k)
+        from sklearn.model_selection import StratifiedKFold as KFold
+
+        kfold = KFold(k, shuffle=shuffle).split(X, Y)
+        accuracies = np.zeros(k)
+        for i, (train_idx, test_idx) in enumerate(
+            tqdm(kfold, total=k, desc="Computing kfold")
+        ):
+            xtrain = [X[i] for i in train_idx]
+            ytrain = [Y[i] for i in train_idx]
+            cl.fit(xtrain, ytrain)
+            xtest = [X[i] for i in test_idx]
+            ytest = [Y[i] for i in test_idx]
+            accuracies[i] = cl.score(xtest, ytest)
+            if verbose:
+                print(f"step {i+1}, {dataset} : {accuracies[i]}", flush=True)
+                try:
+                    print("Best classification parameters : ", cl.best_params_)
+                except:
+                    None
+
+        print(
+            f"""Accuracy {dataset} : {np.mean(accuracies).round(decimals=3)}±{np.std(accuracies).round(decimals=3)}"""
+        )
+    elif k > 0:
+        from sklearn.model_selection import train_test_split
+
+        print("Computing accuracy, with train test split", flush=True)
+        xtrain, xtest, ytrain, ytest = train_test_split(
+            X, Y, shuffle=shuffle, test_size=k
+        )
+        print("Fitting...", end="", flush=True)
+        cl.fit(xtrain, ytrain)
+        print("Computing score...", end="", flush=True)
+        accuracies = cl.score(xtest, ytest)
+        try:
+            print("Best classification parameters : ", cl.best_params_)
+        except:
+            None
+        print("Done.")
+        if verbose:
+            print(f"Accuracy {dataset} : {accuracies} ")
+    file_path: str = f"result_{dataset}.csv".replace("/", "_").replace(".off", "")
+    columns: list[str] = ["dataset", "cv", "mean", "std"]
+    if exists(file_path):
+        df: pd.DataFrame = pd.read_csv(file_path)
+    else:
+        df: pd.DataFrame = pd.DataFrame(columns=columns)
+    more_names = []
+    more_values = []
+    for key, value in more_columns.items():
+        if key not in columns:
+            more_names.append(key)
+            more_values.append(value)
+        else:
+            warn(f"Duplicate key {key} ! with value {value}")
+    new_line: pd.DataFrame = pd.DataFrame(
+        [
+            [
+                dataset,
+                k,
+                np.mean(accuracies).round(decimals=3),
+                np.std(accuracies).round(decimals=3),
+            ]
+            + more_values
+        ],
+        columns=columns + more_names,
+    )
+    print(new_line)
+    df = pd.concat([df, new_line])
+    df.to_csv(file_path, index=False)

multipers/ml/invariants_with_persistable.py

@@ -0,0 +1,79 @@
+import persistable
+
+
+# requires installing ripser (pip install ripser) as well as persistable from the higher-homology branch,
+# which can be done as follows:
+#     pip install git+https://github.com/LuisScoccola/persistable.git@higher-homology
+# NOTE: only accepts as input a distance matrix
+def hf_degree_rips(
+    distance_matrix,
+    min_rips_value,
+    max_rips_value,
+    max_normalized_degree,
+    min_normalized_degree,
+    grid_granularity,
+    max_homological_dimension,
+    subsample_size = None,
+):
+    if subsample_size == None:
+        p = persistable.Persistable(distance_matrix, metric="precomputed")
+    else:
+        p = persistable.Persistable(distance_matrix, metric="precomputed", subsample=subsample_size)
+
+    rips_values, normalized_degree_values, hilbert_functions, minimal_hilbert_decompositions = p._hilbert_function(
+        min_rips_value,
+        max_rips_value,
+        max_normalized_degree,
+        min_normalized_degree,
+        grid_granularity,
+        homological_dimension=max_homological_dimension,
+    )
+
+    return rips_values, normalized_degree_values, hilbert_functions, minimal_hilbert_decompositions
+
+
+
+def hf_h0_degree_rips(
+    point_cloud,
+    min_rips_value,
+    max_rips_value,
+    max_normalized_degree,
+    min_normalized_degree,
+    grid_granularity,
+):
+    p = persistable.Persistable(point_cloud, n_neighbors="all")
+
+    rips_values, normalized_degree_values, hilbert_functions, minimal_hilbert_decompositions = p._hilbert_function(
+        min_rips_value,
+        max_rips_value,
+        max_normalized_degree,
+        min_normalized_degree,
+        grid_granularity,
+    )
+
+    return rips_values, normalized_degree_values, hilbert_functions[0], minimal_hilbert_decompositions[0] 
+
+
+def ri_h0_degree_rips(
+    point_cloud,
+    min_rips_value,
+    max_rips_value,
+    max_normalized_degree,
+    min_normalized_degree,
+    grid_granularity,
+):
+    p = persistable.Persistable(point_cloud, n_neighbors="all")
+
+    rips_values, normalized_degree_values, rank_invariant, _, _ = p._rank_invariant(
+        min_rips_value,
+        max_rips_value,
+        max_normalized_degree,
+        min_normalized_degree,
+        grid_granularity,
+    )
+
+    return rips_values, normalized_degree_values, rank_invariant
+
+
+
+

multipers/ml/kernels.py

@@ -0,0 +1,176 @@
+from sklearn.base import BaseEstimator, TransformerMixin, clone
+import numpy as np
+from typing import Iterable
+
+
+# To do k folds with a distance matrix, we need to slice it into list of distances.
+# k-fold usually shuffles the lists, so we need to add an identifier to each entry,
+#
+class DistanceMatrix2DistanceList(BaseEstimator, TransformerMixin):
+    def __init__(self) -> None:
+        super().__init__()
+
+    def fit(self, X, y=None):
+        return self
+
+    def transform(self, X):
+        X = np.asarray(X)
+        assert X.ndim == 2  # Its a matrix
+        return np.asarray([[i, *distance_to_pt] for i, distance_to_pt in enumerate(X)])
+
+
+class DistanceList2DistanceMatrix(BaseEstimator, TransformerMixin):
+    def __init__(self) -> None:
+        super().__init__()
+
+    def fit(self, X, y=None):
+        return self
+
+    def transform(self, X):
+        index_list = (
+            np.asarray(X[:, 0], dtype=int) + 1
+        )  # shift of 1, because the first index is for indexing the pts
+        return X[:, index_list]  # The distance matrix of the index_list
+
+
+class DistanceMatrices2DistancesList(BaseEstimator, TransformerMixin):
+    """
+    Input (degree) x (distance matrix) or (axis) x (degree) x (distance matrix D)
+    Output _ (D1) x opt (axis) x (degree) x (D2, , with indices first)
+    """
+
+    def __init__(self) -> None:
+        super().__init__()
+        self._axes = None
+
+    def fit(self, X, y=None):
+        X = np.asarray(X)
+        self._axes = X.ndim == 4
+        assert (
+            self._axes or X.ndim == 3
+        ), " Bad input shape. Input is either (degree) x (distance matrix) or (axis) x (degree) x (distance matrix) "
+
+        return self
+
+    def transform(self, X):
+        X = np.asarray(X)
+        assert (X.ndim == 3 and not self._axes) or (
+            X.ndim == 4 and self._axes
+        ), f"X shape ({X.shape}) is not valid"
+        if self._axes:
+            out = np.asarray(
+                [
+                    [
+                        DistanceMatrix2DistanceList().fit_transform(M)
+                        for M in matrices_in_axes
+                    ]
+                    for matrices_in_axes in X
+                ]
+            )
+            return np.moveaxis(out, [2, 0, 1, 3], [0, 1, 2, 3])
+        else:
+            out = np.array(
+                [DistanceMatrix2DistanceList().fit_transform(M) for M in X]
+            )  # indices are at [:,0,Any_coord]
+            # return np.moveaxis(out, 0, -1) ## indices are at [:,0,any_coord], degree axis is the last
+            return np.moveaxis(out, [1, 0, 2], [0, 1, 2])
+
+    def predict(self, X):
+        return self.transform(X)
+
+
+class DistancesLists2DistanceMatrices(BaseEstimator, TransformerMixin):
+    """
+    Input (D1) x opt (axis) x (degree) x (D2 with indices first)
+    Output opt (axis) x (degree) x (distance matrix (D1,D2))
+    """
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.train_indices = None
+        self._axes = None
+
+    def fit(self, X: np.ndarray, y=None):
+        X = np.asarray(X)
+        assert X.ndim in [3, 4]
+        self._axes = X.ndim == 4
+        if self._axes:
+            self.train_indices = np.asarray(X[:, 0, 0, 0], dtype=int)
+        else:
+            self.train_indices = np.asarray(X[:, 0, 0], dtype=int)
+        return self
+
+    def transform(self, X):
+        X = np.asarray(X)
+        assert X.ndim in [3, 4]
+        # test_indices = np.asarray(X[:,0,0], dtype=int)
+        # print(X.shape, self.train_indices, test_indices, flush=1)
+        # First coord of X is test indices by design, train indices have to be selected in the second coord, last one is the degree
+        if self._axes:
+            Y = X[:, :, :, self.train_indices + 1]
+            return np.moveaxis(Y, [0, 1, 2, 3], [2, 0, 1, 3])
+        else:
+            Y = X[
+                :, :, self.train_indices + 1
+            ]  # we only keep the good indices # shift of 1, because the first index is for indexing the pts
+            return np.moveaxis(
+                Y, [0, 1, 2], [1, 0, 2]
+            )  # we put back the degree axis first
+
+        # # out = np.moveaxis(Y,-1,0) ## we put back the degree axis first
+        # return out
+
+
+class DistanceMatrix2Kernel(BaseEstimator, TransformerMixin):
+    """
+    Input : (degree) x (distance matrix) or (axis) x (degree) x (distance matrix) in the second case, axis HAS to be specified (meant for cross validation)
+    Output : kernel of the same shape of distance matrix
+    """
+
+    def __init__(
+        self,
+        sigma: float | Iterable[float] = 1,
+        axis: int | None = None,
+        weights: Iterable[float] | float = 1,
+    ) -> None:
+        super().__init__()
+        self.sigma = sigma
+        self.axis = axis
+        self.weights = weights
+        # self._num_axes=None
+        self._num_degrees = None
+
+    def fit(self, X, y=None):
+        if len(X) == 0:
+            return self
+        assert X.ndim in [3, 4], "Bad input."
+        if self.axis is None:
+            assert X.ndim == 3 or X.shape[0] == 1, "Set an axis for data with axis !"
+            if X.shape[0] == 1 and X.ndim == 4:
+                self.axis = 0
+                self._num_degrees = len(X[0])
+            else:
+                self._num_degrees = len(X)
+        else:
+            assert X.ndim == 4, "Cannot choose axis from data with no axis !"
+            self._num_degrees = len(X[self.axis])
+        if isinstance(self.weights, float) or isinstance(self.weights, int):
+            self.weights = [self.weights] * self._num_degrees
+        assert (
+            len(self.weights) == self._num_degrees
+        ), f"Number of weights ({len(self.weights)}) has to be the same as the number of degrees ({self._num_degrees})"
+        return self
+
+    def transform(self, X) -> np.ndarray:
+        if self.axis is not None:
+            X = X[self.axis]
+        # TODO : pykeops, and full pipeline w/ pykeops
+        kernels = np.asarray(
+            [
+                np.exp(-distance_matrix / (2 * self.sigma**2)) * weight
+                for distance_matrix, weight in zip(X, self.weights)
+            ]
+        )
+        out = np.mean(kernels, axis=0)
+
+        return out