This project explores different deep learning models (ANN, RNN, and LSTM) for detecting clickbait headlines.
The dataset used comes from Kaggle:
Clickbait Dataset
It contains news headlines labeled as:
- 1 β Clickbait
- 0 β Non-clickbait
I implemented and trained three models using Keras (TensorFlow):
- Artificial Neural Network (ANN)
- Recurrent Neural Network (RNN)
- Long Short-Term Memory Network (LSTM)
The models were trained and tested on the dataset. Final accuracies:
- ANN: 0.9598
- RNN: 0.9267
- LSTM: 0.9475
- Clone this repository:
git clone https://github.com/istec-iuc/clickbait-detector.git cd clickbait-detector - Install dependencies:
pip install -r requirements.txt
- Run the notebook to train the models
- ANN, RNN, and LSTM were compared on the same dataset to evaluate their performance.
- Preprocessing included text cleaning, tokenization, and padding before feeding data into the models.