search

cerlymarco / MEDIUM_NoteBook

Repository containing notebooks of my posts on Medium
MIT License
2.08k stars 975 forks source link

Question on label setting in OneClass Neural Network #50

Closed mangata7 closed 1 year ago

mangata7 commented 1 year ago

Thank you for your great work. I have a question about label setting.

In train_generator, I think class "cat" is given label "0", but in test_generator, class "dog" is given label "0" , class "cat" is given label "1". Is this fine if training data and test data has disagreement on label?

cerlymarco commented 1 year ago

Hi,

train_generator is used with wrap_generator that transforms cat to class 1 with one-hot encoding

if u support the project don't forget to leave a star ;-)