InklyAI / src /training /__init__.py
pravinai's picture
Upload folder using huggingface_hub
8eab354 verified
"""
Training package for signature verification.
"""
from .trainer import SignatureTrainer, SignatureDataset, TripletDataset
from .losses import (
ContrastiveLoss,
TripletLoss,
CombinedLoss,
AdaptiveLoss,
FocalLoss
)
__all__ = [
'SignatureTrainer',
'SignatureDataset',
'TripletDataset',
'ContrastiveLoss',
'TripletLoss',
'CombinedLoss',
'AdaptiveLoss',
'FocalLoss'
]