@@ -30,8 +30,8 @@ class Rips(BaseEstimator):
30
30
31
31
Attributes
32
32
----------
33
- _dgm : list of ndarray, each shape (n_pairs, 2)
34
- After `transform`, _dgm contains computed persistence diagrams in
33
+ dgm_ : list of ndarray, each shape (n_pairs, 2)
34
+ After `transform`, dgm_ contains computed persistence diagrams in
35
35
each dimension
36
36
37
37
Examples
@@ -55,7 +55,7 @@ def __init__(self, maxdim=1, thresh=-1, coeff=2):
55
55
self .maxdim = maxdim
56
56
self .thresh = thresh
57
57
self .coeff = coeff
58
- self ._dgm = None
58
+ self .dgm_ = None
59
59
60
60
def transform (self , X , distance_matrix = False , metric = 'euclidean' ):
61
61
"""Compute persistence diagrams for X data array.
@@ -78,7 +78,7 @@ def transform(self, X, distance_matrix=False, metric='euclidean'):
78
78
X = pairwise_distances (X , metric = metric )
79
79
80
80
dgm = self ._compute_rips (X )
81
- self ._dgm = dgm
81
+ self .dgm_ = dgm
82
82
83
83
return dgm
84
84
@@ -87,7 +87,7 @@ def fit_transform(self, X, distance_matrix=False, metric='euclidean'):
87
87
88
88
"""
89
89
self .transform (X , distance_matrix , metric )
90
- return self ._dgm
90
+ return self .dgm_
91
91
92
92
def _compute_rips (self , dm ):
93
93
""" Compute the persistence diagram
@@ -126,7 +126,7 @@ def plot(self, diagram=None, diagonal=True, sz=20, labels='dgm', axcolor=np.arra
126
126
""" Plot each diagram on the same plot.
127
127
"""
128
128
if diagram is None :
129
- diagram = self ._dgm
129
+ diagram = self .dgm_
130
130
131
131
if type (diagram ) is not list :
132
132
diagram = [diagram ]
0 commit comments