I made a csv file that contains the best data science articles from various tags that I scraped using Parsehub on the Medium.com website. The csv file has information on the title of the article, tags used, publication, claps received, number of responses etc. This dataset is available on Kaggle and is called Medium-Search-Dataset.
Our task is to find the coefficients of each equation. To speed up the training time we'll be using a pre-trained neural network Resnet-50. Since this is a regression problem I will be altering the model to better suit our data.
My task is to make a text generator to generate coherent Article Titles. I will be using the transformers library for pre-processing and model building I will then finetune the model using PyTorch Lightning.
This project is a part of my series 25 projects challenge. If you want to see more of these or want to take part yourself, please visit the introduction page.