Add ClassificationDataset API - #125
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
ClassificationDatasetAPI 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.