Skip to content

Commit a78741e

Browse files
authored
Update README.md
1 parent b3c4477 commit a78741e

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

README.md

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,17 @@
11
# Text-Summarization-with-Amazon-Reviews
2-
A seq2seq model that can generate summaries from fine food reviews on Amazon
2+
3+
The objective of this project is to build a seq2seq model that can create relevant summaries for reviews written about fine foods sold on Amazon. This dataset contains above 500,000 reviews, and is hosted on [Kaggle.](https://www.kaggle.com/snap/amazon-fine-food-reviews). It's too large to host here, it's over 300MB.
4+
5+
To build our model we will use a two-layered bidirectional RNN with LSTMs on the input data and two layers, each with an LSTM using bahdanau attention on the target data. [Jaemin Cho's tutorial](https://github.com/j-min/tf_tutorial_plus/tree/master/RNN_seq2seq/contrib_seq2seq) for seq2seq was really helpful to get the code in working order because this is my first project with TensorFlow 1.1; some of the functions are very different from 1.0. The architecture for this model is similar to Xin Pan's and Peter Liu's, here's their [GitHub page.](https://github.com/tensorflow/models/tree/master/textsum)
6+
7+
This model uses [Conceptnet Numberbatch's](https://github.com/commonsense/conceptnet-numberbatch) pre-trained word vectors.
8+
9+
Here are some examples of reviews and their generated summaries:
10+
- Description(1): The coffee tasted great and was at such a good price! I highly recommend this to everyone!
11+
- Summary(1): great coffee
12+
13+
- Description(2): This is the worst cheese that I have ever bought! I will never buy it again and I hope you won’t either!
14+
- Summary(2): omg gross gross
15+
16+
- Description(3): love individual oatmeal cups found years ago sam quit selling sound big lots quit selling found target expensive buy individually trilled get entire case time go anywhere need water microwave spoon know quaker flavor packets
17+
- Summary(3): love it

0 commit comments

Comments
 (0)