Skip to content

A Multi-label Text Classification with BERT, fastai and torch.

Notifications You must be signed in to change notification settings

judytraj007/Multi-label-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 

Repository files navigation

Multi-label-Classification

A Multi-label Text Classification with BERT, fastai and torch.

Given the data set the task is to predict the tags associated with a recipe. Input - A dataframe with the columns:(”cooking_method”,”cuisine”,”ingredients”,”prep_time”,”recipe_name”,”serves”) Target Column - tags

About

A Multi-label Text Classification with BERT, fastai and torch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published