https://github.com/tensorly/tensorly
Tip revision: bdc11743a1e8d283d4b9b12cc1b17930c12b0518 authored by Jean Kossaifi on 07 December 2020, 19:05:18 UTC
DOC: fix class documentation
DOC: fix class documentation
Tip revision: bdc1174
_base_decomposition.py
"""Base classes for all estimators, ensure compatibility with Scikit-Learn
"""
class DecompositionMixin:
def fit(self, tensor, **kwargs):
_ = self.fit_transform(tensor, **kwargs)
return self
def __repr__(self):
return f'{self.__class__.__name__} decomposition of rank {self.rank}.'