Join GitHub today
GitHub is home to over 31 million developers working together to host and review code, manage projects, and build software together.Sign up
Does flow_from_dataframe() support multilabel outputs? #135
I have previously been able to train a CNN to have multi-class and multi-label output using flow(), but I am having trouble getting it to train where the labels are vectors of binary values. I am not sure what class_mode I should be using? @Vijayabhaskar96 is this supported with flow_from_dataframe()?
@scott-zockoll Sorry I was busy didn't notice the issue.
You can perform this easily with Sklearn's MultiLabelBinarizer
You can now pass the
@Vijayabhaskar96 indeed I think that would work. Just something I wanted to point out:
By using the MultiLabelBinarizer or basically expanding the tools you impact the memory footprint of the algorithm without necessity, not only during the expansion process but also the class it self
An option would be to support SparseDataFrame BTW, perhaps too far fetch.
I think what you propose with multiple columns should be the way for multi-output, then we would have a very nice consistent API for