"Cleaned up unsupervised_dataset.py"