diff options
Diffstat (limited to 'src/python/gudhi/sktda/kernel_methods.py')
| -rw-r--r-- | src/python/gudhi/sktda/kernel_methods.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/src/python/gudhi/sktda/kernel_methods.py b/src/python/gudhi/sktda/kernel_methods.py index b8d4ab3a..0a0d333d 100644 --- a/src/python/gudhi/sktda/kernel_methods.py +++ b/src/python/gudhi/sktda/kernel_methods.py @@ -1,6 +1,11 @@ -""" -@author: Mathieu Carriere -All rights reserved +""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT. + See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details. + Author(s): Mathieu Carrière + + Copyright (C) 2018-2019 Inria + + Modification(s): + - YYYY/MM Author: Description of the modification """ import numpy as np @@ -140,7 +145,7 @@ class PersistenceScaleSpaceKernel(BaseEstimator, TransformerMixin): """ self.diagrams_ = list(X) for i in range(len(self.diagrams_)): - op_D = np.matmul(self.diagrams_[i], np.array([[0.,1.], [1.,0.]])) + op_D = self.diagrams_[i][:,[1,0]] self.diagrams_[i] = np.concatenate([self.diagrams_[i], op_D], axis=0) self.pwg_.fit(X) return self |