About this document
IMDB Dataset Neural Network Model by 4416 Likhitha is a document available to read on EtoBox.
The document loads and preprocesses movie review data from the IMDB dataset to build and evaluate a simple neural network movie classifier. It loads and vectorizes the training and test data, defines a model with an input, two hidden and an output layer, trains it for 20 epochs and evaluates its accuracy on the test set, achieving 50% accuracy. It then plots the confusion matrix to analyze errors.
- Author
- 4416 Likhitha
- Language
- EN