Build multiclass and multilabel tf.dataset to train Bert

TensorFlow General Discussion
,
1

Hi there,
I’m searching a way to buil a tf.dataset with not just multiple labels but with multiple classes to, but i didn’t find. Is there a way to build this kind of tf.dataset? i’m trying to build a text generator where i can pass multiple arguments to generate a wine description.

wine_dataset
wine_dataset1342×523 41.6 KB

3

Hi @alencarsjr

Welcome to the TensorFlow Forum!

Please provide some more details on the issue what is the dataset type, model building objective, dataset shape and minimal reproducible code to understand the issue better.

You can refer to the TensorFlow Text tutorials where Text Generation, Text Classification and Bert NLP models are explained. Thank you.