dijkprofile_annotator/training/__init__.py

from .train import train
from .train import get_loss_train
from .train import accuracy_check
from .train import accuracy_check_for_batch