Skip to content

STL dataset returns test fold if fold is misspecified #447

@reynoldscem

Description

@reynoldscem

If you miss-spell, or otherwise pass an invalid fold to the STL dataset it will set the data and labels to that of the test set.

Can be seen here:

else: # self.split == 'test':

Other datasets appear to raise a ValueError in their constructor when passed an invalid fold:

if split not in self.splits:

Would be good to do so for STL-10 also.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions