In this article, we are going to use a dataset based on a popular TV Series “The Big Bang Theory”. We will perform a very basic level Exploratory Data Analysis (EDA) on the dataset and then make a recommendation system also. This article is project-based, and you can follow along by downloading the dataset from priyankad0993/big-band-theory-information | Workspace | data.world
(The dataset is freely available to the public for use. Also, the file is in .xlsx format, you can convert it to csv file by simply saving it in that format).
Let’s Start!!
Data Preparation and Cleaning
You will rarely come across a dataset that you can use as it is i.e. the dataset will be not ready to use, without you transforming.
Let’s first start by importing the required libraries and uploading our dataset. You can start a new Jupyter notebook or use a Google collab notebook for this purpose.
Let’s start by installing and importing Pandas and Numpy libraries of python.
#installing pandas and numpy
!pip install pandas — upgrade — quiet
!pip install numpy — upgrade — quiet
#importing pandas and numpy
import pandas as pd
import numpy as np
It’s good to import pandas as pd and numpy as np because it is easier to use them.
Now let’s read our dataset by using the name of the csv file that we saved. You will need to upload the dataset file first in the files section of the jupyter notebook or google collab.
#reading the dataset
df = pd.read_csv(‘BigBangTheoryIMDB.csv’)
The reason that I named the file as df is that Pandas stores the file as dataframe, which is a data type just like a string, int, etc.
Now let’s see how our data looks by viewing the first few rows of our data using head().
df.head()
screenshot of the output
Now let’s see how many rows and columns do we have in our dataset by using shape().
df.shape()
screenshot of the output
Now let’s see the basic information about the data, like which type is each column, etc.
df.info()
screenshot of the output
If you are thinking about whether you need to memorize all of these various functions like head, shape, etc. then the answer is No, you will know what to use with practice, but for the time being, just search on the internet what you want to achieve with your data and surely you will find an answer.
Whenever you clean your data there are some things that you need to generally do like checking for null values and replacing them with the appropriate value and finding duplicate values and eliminating them.
df.duplicated().sum()
screenshot of the output
df.isna().sum()
screenshot of the output
As you might have seen that in our data Season and Episodes are just individual numbers, I am going to add “S” and “E” in Season and Episode Number, just to make it look a bit more presentable.
It is as simple as concatenating two strings using the ‘+’ operator.
Don’t worry about each line of the above code it just set some basic things like chart size, font size, etc. You can only import the libraries using the first three lines of the code.
Let’s see how many episodes are there per season, for that we will use value_counts(). It returns the total values according to a specified column in this case Season. Sorting is set to false because we want the result season-wise and not in descending order which is what is given by default.
Now let’s create a bar chart for this using seaborn.
sns.barplot(x = episode_per_season.index, y = episode_per_season);
screenshot of the output
Again you can simply search the internet for its syntax until you get the hang of it. Let’s see the names of the top 10 and bottom 10 episodes according to ratings.
There are two types of recommendation system models content-based and collaborative based you can more about them here. But in this model, we will use content-based, which means If you watched the movie “A” and according to the plot movie “B” is similar to “A”, then “B” will be recommended to you to watch.
In our dataset, we are given a plot of each episode, so we are going to use it to check similarity between episodes,
We will import the required scikit libraries for this
from sklearn.feature_extraction.text import CountVectorizer
#stop_words attribute will skip all the helper words that occur english language eg: in, the , they etc.
cv_fvector = CountVectorizer(stop_words=’english’)
vector = cv_fvector.fit_transform(df[‘Plot’]).toarray()
Here we have imported count vectorizer. Since machines can’t understand the text as we do, this library converts the text into numerical data (in this case vectors). How it does that is, it counts the frequency of each distinct word that occurs in a sentence. For instance:
Sample = [“This is cat”, “This is dog”]
And then we used fit_transform to autoscale the data.
from sklearn.metrics.pairwise import cosine_similarity
similarity_array = cosine_similarity(vector)
similarity_array
screenshot of the output
It will give us an array calculating the cosine distance of each vector with the other vector.Let’s test our model
episode = 'S12 E5 The Planetarium Collision'
//this gives the index no. of the episode in our data set
Lastly, we use a very simple for loop to fetch the names of our results
for i in similar_eps[1:6]:
print(df.iloc[i[0]].SE_Title)
screenshot of the output
This was the final result that I got.
Conclusion
We performed some EDA and made a pretty decent recommendation system. You can still do a lot of things with this dataset. I didn’t use the “votes” column, you can use it to generate some useful insights. You can make the “Plot” column, a bit more relevant for the recommendation system, by adding ratings, season number, etc. to it. Just think about what could be possible and there will be a way to achieve the desired result. There are so many awesome datasets available, use them and you will find some very interesting insights. You can download any freely available dataset on movies, books, etc, and make your very own recommendation system. Don’t Stop now.
The media shown in this article is not owned by Analytics Vidhya and are used at the Author’s discretion.
We use cookies essential for this site to function well. Please click to help us improve its usefulness with additional cookies. Learn about our use of cookies in our Privacy Policy & Cookies Policy.
Show details
Powered By
Cookies
This site uses cookies to ensure that you get the best experience possible. To learn more about how we use cookies, please refer to our Privacy Policy & Cookies Policy.
brahmaid
It is needed for personalizing the website.
csrftoken
This cookie is used to prevent Cross-site request forgery (often abbreviated as CSRF) attacks of the website
Identityid
Preserves the login/logout state of users across the whole site.
sessionid
Preserves users' states across page requests.
g_state
Google One-Tap login adds this g_state cookie to set the user status on how they interact with the One-Tap modal.
MUID
Used by Microsoft Clarity, to store and track visits across websites.
_clck
Used by Microsoft Clarity, Persists the Clarity User ID and preferences, unique to that site, on the browser. This ensures that behavior in subsequent visits to the same site will be attributed to the same user ID.
_clsk
Used by Microsoft Clarity, Connects multiple page views by a user into a single Clarity session recording.
SRM_I
Collects user data is specifically adapted to the user or device. The user can also be followed outside of the loaded website, creating a picture of the visitor's behavior.
SM
Use to measure the use of the website for internal analytics
CLID
The cookie is set by embedded Microsoft Clarity scripts. The purpose of this cookie is for heatmap and session recording.
SRM_B
Collected user data is specifically adapted to the user or device. The user can also be followed outside of the loaded website, creating a picture of the visitor's behavior.
_gid
This cookie is installed by Google Analytics. The cookie is used to store information of how visitors use a website and helps in creating an analytics report of how the website is doing. The data collected includes the number of visitors, the source where they have come from, and the pages visited in an anonymous form.
_ga_#
Used by Google Analytics, to store and count pageviews.
_gat_#
Used by Google Analytics to collect data on the number of times a user has visited the website as well as dates for the first and most recent visit.
collect
Used to send data to Google Analytics about the visitor's device and behavior. Tracks the visitor across devices and marketing channels.
AEC
cookies ensure that requests within a browsing session are made by the user, and not by other sites.
G_ENABLED_IDPS
use the cookie when customers want to make a referral from their gmail contacts; it helps auth the gmail account.
test_cookie
This cookie is set by DoubleClick (which is owned by Google) to determine if the website visitor's browser supports cookies.
_we_us
this is used to send push notification using webengage.
WebKlipperAuth
used by webenage to track auth of webenagage.
ln_or
Linkedin sets this cookie to registers statistical data on users' behavior on the website for internal analytics.
JSESSIONID
Use to maintain an anonymous user session by the server.
li_rm
Used as part of the LinkedIn Remember Me feature and is set when a user clicks Remember Me on the device to make it easier for him or her to sign in to that device.
AnalyticsSyncHistory
Used to store information about the time a sync with the lms_analytics cookie took place for users in the Designated Countries.
lms_analytics
Used to store information about the time a sync with the AnalyticsSyncHistory cookie took place for users in the Designated Countries.
liap
Cookie used for Sign-in with Linkedin and/or to allow for the Linkedin follow feature.
visit
allow for the Linkedin follow feature.
li_at
often used to identify you, including your name, interests, and previous activity.
s_plt
Tracks the time that the previous page took to load
lang
Used to remember a user's language setting to ensure LinkedIn.com displays in the language selected by the user in their settings
s_tp
Tracks percent of page viewed
AMCV_14215E3D5995C57C0A495C55%40AdobeOrg
Indicates the start of a session for Adobe Experience Cloud
s_pltp
Provides page name value (URL) for use by Adobe Analytics
s_tslv
Used to retain and fetch time since last visit in Adobe Analytics
li_theme
Remembers a user's display preference/theme setting
li_theme_set
Remembers which users have updated their display / theme preferences
We do not use cookies of this type.
_gcl_au
Used by Google Adsense, to store and track conversions.
SID
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
SAPISID
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
__Secure-#
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
APISID
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
SSID
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
HSID
Save certain preferences, for example the number of search results per page or activation of the SafeSearch Filter. Adjusts the ads that appear in Google Search.
DV
These cookies are used for the purpose of targeted advertising.
NID
These cookies are used for the purpose of targeted advertising.
1P_JAR
These cookies are used to gather website statistics, and track conversion rates.
OTZ
Aggregate analysis of website visitors
_fbp
This cookie is set by Facebook to deliver advertisements when they are on Facebook or a digital platform powered by Facebook advertising after visiting this website.
fr
Contains a unique browser and user ID, used for targeted advertising.
bscookie
Used by LinkedIn to track the use of embedded services.
lidc
Used by LinkedIn for tracking the use of embedded services.
bcookie
Used by LinkedIn to track the use of embedded services.
aam_uuid
Use these cookies to assign a unique ID when users visit a website.
UserMatchHistory
These cookies are set by LinkedIn for advertising purposes, including: tracking visitors so that more relevant ads can be presented, allowing users to use the 'Apply with LinkedIn' or the 'Sign-in with LinkedIn' functions, collecting information about how visitors use the site, etc.
li_sugr
Used to make a probabilistic match of a user's identity outside the Designated Countries
MR
Used to collect information for analytics purposes.
ANONCHK
Used to store session ID for a users session to ensure that clicks from adverts on the Bing search engine are verified for reporting purposes and for personalisation
We do not use cookies of this type.
Cookie declaration last updated on 24/03/2023 by Analytics Vidhya.
Cookies are small text files that can be used by websites to make a user's experience more efficient. The law states that we can store cookies on your device if they are strictly necessary for the operation of this site. For all other types of cookies, we need your permission. This site uses different types of cookies. Some cookies are placed by third-party services that appear on our pages. Learn more about who we are, how you can contact us, and how we process personal data in our Privacy Policy.