Skip to content

NLP Machine Learning pipeline written in Python on Jupyter Notebook, using the PyTorch library. This neural network is built with transformer architecture functions to transform internet reviews into stylized poems. It does so by taking a review as an input, summarizes the review, and generates poems using the shortened summary.

Notifications You must be signed in to change notification settings

davidtran001/Poetify

 
 

Repository files navigation

Poetify

Poetify is a NLP Machine Learning pipeline written in Python on Jupyter Notebook, using the PyTorch library. This neural network is built with transformer architecture functions to transform internet reviews into stylized poems. It does so by taking a review as an input, summarizes the review, and generates poems using the shortened summary.

Web Application

Users can try out Poetify using the Poetify web application built with Anvil.

Contributors

This project was co-developed by David Tran (www.github.com/davidtran001), Stephen Wang (www.github.com/Stephenwang3801), William Sati (www.github.com/WilliamSati), and Percy Wang (www.github.com/junwjie).

For more information regarding the project regarding models, data processing, and architecture please click the link below: https://github.com/Stephenwang3801/Poetify/blob/main/Poetify%20Desc.pdf

Model Architecture
Example of what the model does:
Context Transfer
Highlighted text represents the text in the input that had their subject and sentiment transferred to the output poem. Blue represents how the music subject, text highlighted in red represents the calm and quiet sentiment.

Datasets

Poetify was trained on various review and poem datasets sourced from Kaggle

Haiku Dataset

Complete poetryfoundation.com dataset

Amazon Reviews for Sentiment Analysis

About

NLP Machine Learning pipeline written in Python on Jupyter Notebook, using the PyTorch library. This neural network is built with transformer architecture functions to transform internet reviews into stylized poems. It does so by taking a review as an input, summarizes the review, and generates poems using the shortened summary.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 97.5%
  • Python 2.5%