import os
import sys
import random
from pathlib import Path

sys.path.insert(0, '../src')
from utils import collate_samples, evaluate_samples, show_samples_librimix
from lib.utils import setup_determinism

EXPERIMENTS_ROOT = Path('/ssd/ertug/experiments')
LIBRIMIX_ROOT = '/ssd/datasets/librimix'

setup_determinism(0)

LibriMix Samples (MixIT)

data_samples = collate_samples(
    dataset_name='librimix',
    dataset_root=LIBRIMIX_ROOT,
    count=10
)
separation_samples = evaluate_samples(
    train_results_root=EXPERIMENTS_ROOT / 'librimix_100p',
    dataset_name='librimix',
    model_name='mixit',
    data_samples=data_samples,
    mixcycle_init_epochs=50,
    train_subsample_ratio=1.0,
    eval_epochs=1,
    run_id=0
)
show_samples_librimix(separation_samples)