1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
from .. import CubicalComplex
from sklearn.base import BaseEstimator, TransformerMixin
# joblib is required by scikit-learn
from joblib import Parallel, delayed
class CubicalPersistence(BaseEstimator, TransformerMixin):
# Fast way to find primes and should be enough
_available_primes = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97]
"""
This is a class for computing the persistence diagrams from a cubical complex.
"""
def __init__(self, dimensions=None, persistence_dim=0, min_persistence=0, n_jobs=None):
"""
Constructor for the CubicalPersistence class.
Parameters:
dimensions (list of int): A list of number of top dimensional cells.
persistence_dim (int): The returned persistence diagrams dimension. Default value is `0`.
min_persistence (float): The minimum persistence value to take into account (strictly greater than
`min_persistence`). Default value is `0.0`. Sets `min_persistence` to `-1.0` to see all values.
n_jobs (int): cf. https://joblib.readthedocs.io/en/latest/generated/joblib.Parallel.html
"""
self.dimensions = dimensions
self.persistence_dim = persistence_dim
self.homology_coeff_field_ = None
for dim in self._available_primes:
if dim > persistence_dim + 1:
self.homology_coeff_field_ = dim
break
if self.homology_coeff_field_ == None:
raise ValueError("persistence_dim must be less than 96")
self.min_persistence = min_persistence
self.n_jobs = n_jobs
def fit(self, X, Y=None):
"""
Nothing to be done.
"""
return self
def __transform(self, cells):
cubical_complex = CubicalComplex(top_dimensional_cells=cells, dimensions=self.dimensions)
cubical_complex.compute_persistence(
homology_coeff_field=self.homology_coeff_field_, min_persistence=self.min_persistence
)
diagrams = cubical_complex.persistence_intervals_in_dimension(self.persistence_dim)
return diagrams
def transform(self, X, Y=None):
"""
Compute all the cubical complexes and their associated persistence diagrams.
Parameters:
X (list of list of double OR list of numpy.ndarray): List of cells filtration values.
Returns:
Persistence diagrams
"""
# threads is preferred as cubical construction and persistence computation releases the GIL
return Parallel(n_jobs=self.n_jobs, prefer="threads")(delayed(self.__transform)(cells) for cells in X)
|