lightning-flash Multilabel

[jorgeutd]

Hello lightning-flash team,

Does flash support multilabel text classification? I tried a couple of times with the target column in multiples formats and it did not work.

Thanks,

Jorge

[ethanwharris]

Hi @jorgeutd thanks for the question! This looks like a bug as it was expected to work. I've opened a couple of issues:

• Multilabel text classification not working #397 A bug issue to get multilabel text working properly
• Multilabel text classification example #396 A feature issue to add an example showing multilabel text classification in action

I'll get back to you here once these are done. Hope that helps 😃

[ethanwharris]

Hi @jorgeutd - as of #401 this is now supported 😃 Take a look at the docs here: https://lightning-flash.readthedocs.io/en/latest/reference/text_classification_multi_label.html

[ethanwharris]

Hi Jorge,

Can I ask what version of Flash you are using? This support is only available on master at the moment, so you would need to install with:

pip install git+https://github.com/PytorchLightning/lightning-flash.git@master#egg=lightning-flash[text]

Hope that helps 😃

[jorgeutd]

Ethan,

This was the issue. I am now able to reproduce the example with any issues. Also, I was looking into the flash doc and I saw that for a summarization tasks the target_max_length argument is available, do you have anything like this for a classification tasks or what is the default max_length for the input text 512?

I am getting the following error now for my custom data:

Has to be the way the csv is encoding the text input field. What would you recommend here:

Again, thank you so much Ethan.

[jorgeutd]

Ethan,

Is there another option to pass the data into the dataloader for this type of tasks? Where I could find examples?

Thanks,

Jorge

[jorgeutd]

Ethan,

Everything is working now. Thank you so much.

[ethanwharris]

@jorgeutd Awesome! Glad you got it working. Please don't hesitate to reach out again if you have any feedback 😃