* add train_idx/val_idx/test_idx as dataset attributes * refine docstring
Attach a file by drag & drop or click to upload
