diff options
author | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-06-24 16:19:34 +0200 |
---|---|---|
committer | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-06-24 16:19:34 +0200 |
commit | 370c09100d94dc73f582ebbabb994bcd2a3820eb (patch) | |
tree | daae040b8079870772dffd7ab24a54d7b1b72387 /src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py | |
parent | b93fbe5e7c246a6ee23a8686f9b5983624a6ab42 (diff) |
changed dimensions into homology_dimensions
Diffstat (limited to 'src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py')
-rw-r--r-- | src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py index 4ec3f7c7..5a8e5b75 100644 --- a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py +++ b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py @@ -43,18 +43,18 @@ class LowerStarSimplexTreeLayer(tf.keras.layers.Layer): """ TensorFlow layer for computing lower-star persistence out of a simplex tree """ - def __init__(self, simplextree, dimensions, min_persistence=None, homology_coeff_field=11, **kwargs): + def __init__(self, simplextree, homology_dimensions, min_persistence=None, homology_coeff_field=11, **kwargs): """ Constructor for the LowerStarSimplexTreeLayer class Parameters: simplextree (gudhi.SimplexTree): underlying simplex tree. Its vertices MUST be named with integers from 0 to n-1, where n is its number of vertices. Note that its filtration values are modified in each call of the class. - dimensions (List[int]): homology dimensions + homology_dimensions (List[int]): list of homology dimensions min_persistence (List[float]): minimum distance-to-diagonal of the points in the output persistence diagrams (default None, in which case 0. is used for all dimensions) homology_coeff_field (int): homology field coefficient. Must be a prime number. Default value is 11. Max is 46337. """ super().__init__(dynamic=True, **kwargs) - self.dimensions = dimensions + self.dimensions = homology_dimensions self.simplextree = simplextree self.min_persistence = min_persistence if min_persistence != None else [0. for _ in range(len(self.dimensions))] self.hcf = homology_coeff_field |