Skip to content

Latest commit

 

History

History
32 lines (22 loc) · 1.98 KB

README.md

File metadata and controls

32 lines (22 loc) · 1.98 KB

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