https://github.com/sovit-123/imdb_neural_net_review_classification
https://github.com/sovit-123/imdb_neural_net_review_classification
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/sovit-123/imdb_neural_net_review_classification
- Owner: sovit-123
- Created: 2019-04-29T05:50:01.000Z (about 7 years ago)
- Default Branch: master
- Last Pushed: 2019-04-29T06:02:59.000Z (about 7 years ago)
- Last Synced: 2025-02-03T10:46:21.084Z (over 1 year ago)
- Language: Jupyter Notebook
- Size: 34.2 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# imdb_neural_net_review_classification
This is movie review classification with the IMDB dataset using neural networks in Keras.
## Dependencies
* Keras
* Numpy
* Matplotlib
**The model scores about 88% accuracy. It can surely be improved. Some steps to move this project further:**
1. More/less number of nodes in the hidden layers (base results with 16).
2. More/less number of hidden layers (base results with 2).
3. Using some other loss function (hint: `mse`) (base results with `binary_crossentropy`).
4. Using different activation function (hint: `tanh`) (base results with `relu`).
**The file is in .ipynb format. Simply clone the repository or download the file to run and reproduce the results.**