https://github.com/KeeganFernandesWork/Medium-Title-Generator

A model which generates data science article Titles
https://github.com/KeeganFernandesWork/Medium-Title-Generator

Last synced: about 1 month ago
JSON representation

A model which generates data science article Titles

• Host: GitHub
• URL: https://github.com/KeeganFernandesWork/Medium-Title-Generator
• Owner: KeeganFernandesWork
• Created: 2021-08-30T05:07:33.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-08-30T05:33:24.000Z (over 3 years ago)
• Last Synced: 2024-10-05T11:02:04.866Z (3 months ago)
• Language: Jupyter Notebook
• Size: 22.5 KB
• Stars: 11
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• StarryDivineSky - Aristotle609/Medium-Title-Generator

README

        
# Title Generator 

## Introduction

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](https://medium.com/r/?url=https%3A%2F%2Fwww.kaggle.com%2Faristotle609%2Fmediumsearchdataset).

## The Problem

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.

## The Process

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.

![](https://cdn-images-1.medium.com/max/800/1*YUHMEcrudbDs7nkngtny7g.png)

## Endnote

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](https://medium.com/r/?url=https%3A%2F%2Fkeeganfdes03.medium.com%2Ff2150afe053%3Fsource%3Dfriends_link%26sk%3D5ae10ad6072aa80c0f1ed3865a0196a8).