Model description This model is t5 base fine tuned on the 190k Medium Articles dataset for predicting article tags using the article textual content as input. While usually formulated as a multi label classification problem, this model deals with tag generation as a text2text generation task (inspiration from text2tags). How to use the model Data cleaning The dataset is composed of Medium articles and their tags. However, each Medium article can have at most five tags, therefore the author needs to choose what he/she believes are the best tags (mainly for SEO related purposes). This means that an article with the "Python" tag may have not the "Programming Languages" tag, even though the first implies the latter. To clean the dataset accounting for this problem, a hand made taxonomy of about 1000 tags was built. Using the taxonomy, the tags of each articles have been augmented (e.g. an article with the "Python" tag will have the "Programming Languages" tag as well, as the taxonomy says that "Python" is part of "Programming Languages"). The taxonomy is not public, if you are interested in it please send an email at chiusanofabio94@gmail.com. Training and evaluation data The model has…
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy