13 lines
537 B
Python
13 lines
537 B
Python
#TODO: License or such
|
|
from .base_projection_layer import BaseProjectionLayer
|
|
from .identity_projection_layer import IdentityProjectionLayer
|
|
from .frob_projection_layer import FrobeniusProjectionLayer
|
|
from .w2_projection_layer import WassersteinProjectionLayer
|
|
|
|
try:
|
|
import cpp_projection
|
|
except ModuleNotFoundError:
|
|
print('[MSB] ITPAL is not installed; KL projections not avaible.')
|
|
from .base_projection_layer import ExceptionProjectionLayer as KLProjectionLayer
|
|
else:
|
|
from .kl_projection_layer import KLProjectionLayer |