Steven Liu comments

Repositories
Issues
Comments

Results 2 comments of


                                            Steven Liu

Error：too many values to unpack (expected 2)

Hi, I've never seen this issue, but I suspect that this is due to sklearn versioning issues. The installation requirements in the README should be enough, but I will take...

training with custom dataset

Yes, changing the config by changing data['type'] and changing the number of classes to 1 should do it. The class that you write should be a torch.data.Dataset class.