This project focuses on building a machine learning model to predict the number of YouTube subscribers for a channel based on historical data and various growth metrics. The model helps estimate future subscriber counts and can assist creators and analysts in tracking audience growth trends.
With the rapid rise of social media platforms like YouTube, predicting subscriber growth has become valuable for content strategy and marketing. Given the huge volume of content and creators, it is essential to forecast subscriber growth to make data-driven decisions. This project aims to develop a model that accurately predicts subscriber count based on historical and other relevant channel features.
- Use a cleaned and preprocessed dataset containing YouTube channel metrics.
- Apply regression-based deep learning models to predict subscriber growth.
- Evaluate model performance using Mean Squared Error (MSE) and Accuracy.
- Visualize the prediction performance using plots.
- Model: Dense Neural Network (Regression)
- Optimizer: Adam
- Loss Function: Mean Squared Error (MSE)
- Evaluation Metrics: Accuracy, MSE
- Language: Python
- Libraries: NumPy, Pandas, Matplotlib, Seaborn, Scikit-learn, TensorFlow / Keras
- Training Accuracy: ~99.9%
- Final MSE: ~422
- The model performs well even for large-scale prediction (in millions), with acceptable prediction differences.
π¦ youtube-subscriber-prediction
βββ π Results/
β βββ Actual vs Predicted.png
β βββ Training vs Validation Loss.png
β βββ Distribution of Residuals.png
βββ π requirements.txt
βββ π README.md
βββ π subscriber_prediction.ipynb
Follow these steps to set up and run the project locally:
-
Clone the repository
git clone https://github.com/Saral64/youtube-subscriber-prediction.git cd youtube-subscriber-prediction -
Install dependencies
pip install -r requirements.txt
-
Run the Jupyter Notebook
jupyter notebook
Open
subscriber_prediction.ipynbin the Jupyter interface to explore and run the code.
- Add more updated YouTube data (2025+)
- Integrate real-time analytics using YouTube APIs
- Apply more advanced models (e.g., LSTM, XGBoost)
- Expand to prediction across multiple social platforms (Instagram, Twitter)