This project is an IMDb Rating Predictor that uses a TabNet Regressor model to predict the IMDb rating of a movie based on its characteristics. The prediction is powered by a Streamlit-based web application, where users can input movie details like runtime, metascore, and number of votes to get an estimated IMDb rating.
- 📊 Machine Learning Model: Uses a trained TabNet Regressor for predictions.
- 🎥 Movie Data Input: Users can enter movie runtime, metascore, votes, and select genre, director, and cast.
- ⚡ Fast Predictions: The app instantly provides an estimated IMDb rating based on user input.
- 🎭 Interactive UI: Built with Streamlit for a seamless user experience.
- 📂 Pre-trained Model: The model is pre-trained on IMDb’s Top 1000 Movies dataset.
The model is trained on the IMDb Top 1000 Movies dataset, which includes details such as:
- 🎬 Movie Title
- ⏳ Runtime
- 🏆 Meta Score (Critic reviews)
- 🗳️ Number of Votes
- 🎭 Genre
- 🎬 Director & Cast
- ⭐ IMDb Rating (Target variable)
- Algorithm: TabNet Regressor (PyTorch-based Deep Learning Model)
- Training Framework: PyTorch + TabNet
- Input Features:
RuntimeMeta ScoreNumber of Votes
- Target Variable: IMDb Rating
- Evaluation Metric: Mean Absolute Error (MAE)
git clone https://github.com/yourusername/imdb-rating-predictor.git
cd imdb-rating-predictorpip install -r requirements.txtstreamlit run app.py- User Inputs Movie Details 🎬
- The Model Processes Inputs 🧠
- Features are Scaled using StandardScaler ⚖️
- TabNet Model Predicts IMDb Rating ⭐
- The Web App Displays Results 📊
- Movie: "Inception"
- Runtime: 148 minutes
- Metascore: 74
- Number of Votes: 2,000,000+
- Predicted IMDb Rating: ~8.8⭐
This project is open-source and available under the MIT License.
Developed by Chandru S 👨💻✨ | AI & Web Developer | Passionate about ML & AI
📩 Feel free to reach out for collaborations!