This project is a machine learning application that predicts the gross income of a movie based on its budget, IMDb score, and Facebook likes.
The project consists of two main components:
-
A linear regression model ( MoviePredict_LinearRegressionModel.pkl ) that is trained on a dataset of movies with their budget, IMDb score, Facebook likes, and gross income.
-
A REST API that exposes the model as a web service using FastAPI framework and pedantic for validation. The API has one endpoint (http://192.168.2.103:8000/predict) that accepts a POST request with a JSON body that contains the budget, IMDb score, and Facebook likes of a movie. Like:
{
"budget": 180000000,
"imdb_score": 8.5,
"facebook_likes": 356789
} -
The API returns a JSON response with the predicted gross income of the movie. Like:
{
"gross_prediction": 500000.98
}