You are on the data science team of a medical research company that’s interested in finding better ways to predict myopia, or nearsightedness. Your team has tried—and failed—to improve their classification model when training on the whole dataset. However, they believe that there might be distinct groups of patients that would be better to analyze separately. So, your supervisor has asked you to explore this possibility by using unsupervised learning.
You have been provided with raw data, so you’ll first need to process it to fit the machine learning models. You will use several clustering algorithms to explore whether the patients can be placed into distinct groups. Then, you’ll create a visualization to share your findings with your team and other key stakeholders.
This activity is broken down into four parts:
-
Part 1: Prepare the Data.
-
Part 2: Apply Dimensionality Reduction.
-
Part 3: Perform a Cluster Analysis with K-means.
-
Part 4: Make a Recommendation.
- Read
myopia.csv
into a Pandas DataFrame.
- Remove the "MYOPIC" column from the dataset.
- Standardize your dataset so that columns that contain larger values do not influence the outcome more than columns with smaller values.
- Perform dimensionality reduction with PCA. How did the number of the features change?
- After performing dimensionality reduction with PCA on the myopia dataset, the number of features was reduced by four, from the original number of features in the dataset. This technique reduced the dimensionality of the data while retaining most of the variance in the dataset, resulting in a more manageable set of features for subsequent analysis.
- Further reduce the dataset dimensions with t-SNE and visually inspect the results. To do this, run t-SNE on the principal components, which is the output of the PCA transformation.
- Create a scatter plot of the t-SNE output. Are there distinct clusters?
- The scatter plot of the t-SNE output did not reveal any apparent clusters in the myopia dataset. While t-SNE is a powerful tool for visualizing high-dimensional data, it is possible that the underlying structure of the dataset is not well-suited for clustering or that the t-SNE algorithm did not identify the clusters effectively. Therefore, further analysis with other unsupervised machine learning techniques is required to identify potential clusters in the dataset.
Create an elbow plot to identify the best number of clusters. Make sure to do the following:
- Use a
for
loop to determine the inertia for eachk
between 1 through 10.
- If possible, determine where the elbow of the plot is, and at which value of
k
it appears.
To visualize the clusters a custom function get_clusters()
was created to perform K-Means clustering and assign each data point to a cluster. The resulting clustered data was stored in a new DataFrame called cluster_data
, which contained the t-SNE features and class labels from the original dataset. The get_clusters()
function was called with k=3
and cluster_data
as input to perform the K-Means clustering. The resulting clustered data was then visualized using a scatter plot that displayed the t-SNE features on the x- and y-axes, with predicted cluster labels color-coded as points.
Based on your findings, write up a brief (one or two sentences) recommendation for your supervisor in your Jupyter Notebook. Can the patients be clustered? If so, into how many clusters?
- Based on my analysis, it appears that the patients can be clustered into three distinct groups using K-Means clustering on the t-SNE features. I arrived at this number by plotting an elbow curve of the inertia values for each k, which indicated that the elbow point is at k=3. Therefore, I recommend clustering the patients into three groups for further analysis.
Reduced dataset from Orinda Longitudinal Study of Myopia conducted by the US National Eye Institute