Trixy3/trainer/voice_recognition/__init__.py

"""
Voice recognition trainer package.

This package provides specialized training capabilities for speaker recognition
and voice identification using state-of-the-art architectures like ECAPA-TDNN,
TitaNet-S, and SpeakerNet-M.
"""

from .models import ECAPA_TDNN, TitaNet_S, SpeakerNet_M, create_voice_recognition_model
from .trainer import VoiceRecognitionTrainer
from .data import VoiceRecognitionDataset, VoiceRecognitionDataPreprocessor

__all__ = [
    "ECAPA_TDNN",
    "TitaNet_S", 
    "SpeakerNet_M",
    "create_voice_recognition_model",
    "VoiceRecognitionTrainer",
    "VoiceRecognitionDataset",
    "VoiceRecognitionDataPreprocessor"
]