cleaner setup for forgiving layers and sparsity, use fixed train and test files for easier reproducibility