data.load_data(): Change label_col to be optional and add class_weights

georgian-io / Multimodal-Toolkit

Multimodal model for text and tabular data with HuggingFace transformers as building block for text data

https://multimodal-toolkit.readthedocs.io

Apache License 2.0

587 stars 84 forks source link

data.load_data(): Change label_col to be optional and add class_weights #61

Closed DougTrajano closed 9 months ago

DougTrajano commented 9 months ago

In this PR, we are moving the label_col argument in the load_data() function to be optional, it allows the usage of this function in training and inference use cases. The TorchTabularTextDataset() accepts the labels as None.

We are also adding the class_weights argument, which is accepted in the TorchTabularTextDataset().

akashsaravanan-georgian commented 9 months ago

Thank you @DougTrajano!