__init__.py

from .metric import MSE, SAD, Grad, Conn

metrics_class_dict = {'sad': SAD, 'mse': MSE, 'grad': Grad, 'conn': Conn}