dataset .ipynb_checkpoints weights predictions checkpoints __pycache__