import os
import sys
import random
from pathlib import Path

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

EXPERIMENTS_ROOT = Path('/home/ertug/experiments')
REALM_ROOT = '/home/ertug/datasets/REAL-M-v0.1.0'

setup_determinism(0)

REAL-M Samples

---

Model A was trained with PIT-DM (supervised) on the LibriMix dataset.

Model B was trained with MixCycle (unsupervised) on the REAL-M dataset.

---

data_samples = collate_samples(
    dataset_name='realm',
    dataset_root=REALM_ROOT,
    count=10
)
separation_samples_librimix = evaluate_samples(
    train_results_root=EXPERIMENTS_ROOT / 'librimix_100p',
    dataset_name='librimix',
    model_name='pit-dm',
    data_samples=data_samples,
    mixcycle_init_epochs=50,
    train_subsample_ratio=1.0,
    eval_epochs=1,
    run_id=0
)
separation_samples_realm = evaluate_samples(
    train_results_root=EXPERIMENTS_ROOT / 'realm',
    dataset_name='realm',
    model_name='mixcycle',
    data_samples=data_samples,
    mixcycle_init_epochs=0,
    eval_method='blind',
    eval_blind_num_repeat=20,
    eval_epochs=40
)
show_samples_realm(separation_samples_librimix, separation_samples_realm)