see model_BalancedDatasets.py