-
Notifications
You must be signed in to change notification settings - Fork 2.1k
Add ClassificationDataset API #125
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
A summary:
|
@SkalskiP I have responded to a lot of your feedback in my latest commit. Please review. I am still confused about this comment:
Can you outline in pseudocode the expected way this function would work? |
@capjamesg ds = ClassificationDataset(...)
ds_train, ds_test = ds.split(split_ratio=0.8)
ds_test, ds_val = ds_test.split(split_ratio=0.5)
ds_train.as_multiclass_folder_structure(...)
ds_test.as_multiclass_folder_structure(...)
ds_val.as_multiclass_folder_structure(...) Notice that |
My latest commit incorporates all your feedback. I left a comment in response to one of your comments re: the class_id [0] issue. |
My latest commits add:
|
I have incorporated all of your feedback into my latest commit. We now have five test cases for the |
Description
This PR introduces a new
ClassificationDataset
API for use in working with classification datasets using supervision. Users can:Type of change
How has this change been tested, please provide a testcase or example of how you tested the change?
I will add a test case once we have discussed the API.
Docs
I will add the requisite docs once we have discussed the API.