batches/ venv/ train_data/ __pycache__/