From d43293fc7ce82315496bcecaa9359dd333bc0745 Mon Sep 17 00:00:00 2001 From: Marc Glisse Date: Sun, 20 Nov 2022 22:44:38 +0100 Subject: Keep perslay-related stuff under gudhi.tensorflow.perslay --- src/python/doc/representations.rst | 22 +++++++++++----------- src/python/gudhi/tensorflow/__init__.py | 3 +-- 2 files changed, 12 insertions(+), 13 deletions(-) diff --git a/src/python/doc/representations.rst b/src/python/doc/representations.rst index 37d3aa53..5686974a 100644 --- a/src/python/doc/representations.rst +++ b/src/python/doc/representations.rst @@ -63,18 +63,18 @@ PersLay import tensorflow as tf from sklearn.preprocessing import MinMaxScaler import gudhi.representations as gdr - import gudhi.tensorflow as gdtf + import gudhi.tensorflow.perslay as prsl diagrams = [np.array([[0.,4.],[1.,2.],[3.,8.],[6.,8.]])] diagrams = gdr.DiagramScaler(use=True, scalers=[([0,1], MinMaxScaler())]).fit_transform(diagrams) diagrams = tf.RaggedTensor.from_tensor(tf.constant(diagrams, dtype=tf.float32)) rho = tf.identity - phi = gdtf.GaussianPerslayPhi((5, 5), ((-.5, 1.5), (-.5, 1.5)), .1) - weight = gdtf.PowerPerslayWeight(1.,0.) + phi = prsl.GaussianPerslayPhi((5, 5), ((-.5, 1.5), (-.5, 1.5)), .1) + weight = prsl.PowerPerslayWeight(1.,0.) perm_op = tf.math.reduce_sum - perslay = gdtf.Perslay(phi=phi, weight=weight, perm_op=perm_op, rho=rho) + perslay = prsl.Perslay(phi=phi, weight=weight, perm_op=perm_op, rho=rho) vectors = perslay(diagrams) print(vectors) @@ -145,41 +145,41 @@ Metrics PersLay ------- -.. autoclass:: gudhi.tensorflow.Perslay +.. autoclass:: gudhi.tensorflow.perslay.Perslay :members: :special-members: :show-inheritance: Weight functions ^^^^^^^^^^^^^^^^ -.. autoclass:: gudhi.tensorflow.GaussianMixturePerslayWeight +.. autoclass:: gudhi.tensorflow.perslay.GaussianMixturePerslayWeight :members: :special-members: :show-inheritance: -.. autoclass:: gudhi.tensorflow.GridPerslayWeight +.. autoclass:: gudhi.tensorflow.perslay.GridPerslayWeight :members: :special-members: :show-inheritance: -.. autoclass:: gudhi.tensorflow.PowerPerslayWeight +.. autoclass:: gudhi.tensorflow.perslay.PowerPerslayWeight :members: :special-members: :show-inheritance: Phi functions ^^^^^^^^^^^^^ -.. autoclass:: gudhi.tensorflow.FlatPerslayPhi +.. autoclass:: gudhi.tensorflow.perslay.FlatPerslayPhi :members: :special-members: :show-inheritance: -.. autoclass:: gudhi.tensorflow.GaussianPerslayPhi +.. autoclass:: gudhi.tensorflow.perslay.GaussianPerslayPhi :members: :special-members: :show-inheritance: -.. autoclass:: gudhi.tensorflow.TentPerslayPhi +.. autoclass:: gudhi.tensorflow.perslay.TentPerslayPhi :members: :special-members: :show-inheritance: diff --git a/src/python/gudhi/tensorflow/__init__.py b/src/python/gudhi/tensorflow/__init__.py index fe01b9cc..1599cf52 100644 --- a/src/python/gudhi/tensorflow/__init__.py +++ b/src/python/gudhi/tensorflow/__init__.py @@ -1,6 +1,5 @@ from .cubical_layer import CubicalLayer from .lower_star_simplex_tree_layer import LowerStarSimplexTreeLayer from .rips_layer import RipsLayer -from .perslay import * -__all__ = ["Perslay", "GridPerslayWeight", "GaussianMixturePerslayWeight", "PowerPerslayWeight", "GaussianPerslayPhi", "TentPerslayPhi", "FlatPerslayPhi", "LowerStarSimplexTreeLayer", "RipsLayer", "CubicalLayer"] +__all__ = ["LowerStarSimplexTreeLayer", "RipsLayer", "CubicalLayer"] -- cgit v1.2.3