Predicting Tags for a Question posted on Stack Exchange using a pre-trained BERT model from Hugging Face and PyTorch Lightning

Stack Exchange Logo
Stack Exchange Logo
Stack Exchange

Stack Exchange is a network of 176 communities that are created and run by experts and enthusiasts who are passionate about a specific topic. The website primarily serves as a platform for users to ask and answer questions.

All questions are tagged with their subject areas. Each can have up to 5 tags since a question might be related to several subjects. Tags make it easy to find questions in your area of interest and enable you to either learn…

Prasad Nageshkar

Applied Natural Language Processing, Machine Learning

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store