metastable-projections-public/metastable_projections/projections/__init__.py

6 lines
249 B
Python
Raw Permalink Normal View History

2022-10-27 21:12:58 +02:00
#TODO: License or such
from .base_projection_layer import BaseProjectionLayer
from .frob_projection_layer import FrobeniusProjectionLayer
from .kl_projection_layer import KLProjectionLayer
from .w2_projection_layer import WassersteinProjectionLayer