Diversified Stock Portfolio Using Clustering Analysis
Overview
This project constructs diversified stock portfolios from the S&P 500 using unsupervised learning. Historical price data is used to compute risk and return features for each stock; K-means clustering groups stocks with similar behavior. Portfolio construction then selects top stocks by Sharpe ratio from each cluster to achieve diversification across clusters and validate performance via backtesting against the S&P 500 index.
Data & Features
Data sources: S&P 500 constituent list and historical stock prices (US equity data fetched via the pipeline). The first 70% of the history is used for model building; the remaining 30% is reserved for validation.
Features used for clustering (all derived from historical data):
Clustering Features
Correlation with S&P 500 index (price correlation).
Beta: sensitivity of stock returns to index returns.
Annualized return (from daily returns, 252-day year).
Annualized volatility (standard deviation of daily returns, annualized).
Sharpe ratio (annualized return / annualized volatility).
Daily change in price (open-to-close) and daily variation (high-to-low), annualized.
Features are scaled (z-score) before clustering. The optimal number of clusters is chosen using the within-cluster sum of squares (elbow method); we use K = 4.
Clustering
K-means is run on the normalized feature matrix with multiple random starts. Cluster membership ensures the portfolio spans different behavior groups rather than concentrating in one segment of the risk/return space.
Portfolio Construction
Two portfolio variants are built:
(1) Diversified by cluster: Within each cluster, stocks are ranked by Sharpe ratio; the top 5 from each of the 4 clusters form a 20-stock portfolio (equal weight per stock).
(2) Top-20 by Sharpe: The top 20 stocks by Sharpe ratio across the full universe, without cluster constraints.
Both are equal-weighted. The diversified-by-cluster portfolio reduces concentration in a single risk/return profile.
Validation & Backtesting
Validation uses the holdout period (last 30% of the data). Daily returns are computed for the portfolio (equal-weighted average of constituent returns) and for the S&P 500 index.
Cumulative returns are plotted for the cluster-based portfolio, the top-20 Sharpe portfolio, and the S&P 500 to compare risk-adjusted performance.
Results
Summary
Cluster statistics
| Cluster | Count | Mean return | Mean vol | Mean Sharpe | Mean beta |
|---|---|---|---|---|---|
| 1 | 96 | 8.30% | 25.70% | 0.34 | 0.79 |
| 2 | 44 | -5.80% | 30.60% | -0.18 | 0.79 |
| 3 | 37 | 32.30% | 31.70% | 1.03 | 1.03 |
| 4 | 19 | -2.00% | 53.80% | -0.03 | 1.72 |
Elbow plot: within-cluster sum of squares
What this shows: Within-cluster sum of squares (WSS) by candidate cluster count K.
How to read it: Look for the elbow where WSS improvement starts flattening; that point suggests a practical K before diminishing returns.
Return vs volatility by cluster
Training-sample K-means assignment (K = 4). Color is cluster; hover for ticker.
Cluster-wise metrics
Mean annualized return, volatility, and Sharpe ratio by cluster.
Mean return
Mean volatility
Mean Sharpe ratio
Feature correlation matrix
Correlation between clustering features (used for K-means). Helps check redundancy.
| ann_return | ann_vol | ann_sharpe… | ann_daily_… | ann_daily_… | beta | cor | |
|---|---|---|---|---|---|---|---|
| ann_return | 1.00 | -0.06 | 0.94 | -0.69 | -0.04 | 0.01 | 0.67 |
| ann_vol | -0.06 | 1.00 | -0.20 | 0.20 | 0.98 | 0.82 | -0.17 |
| ann_sharpe… | 0.94 | -0.20 | 1.00 | -0.68 | -0.18 | -0.09 | 0.70 |
| ann_daily_… | -0.69 | 0.20 | -0.68 | 1.00 | 0.23 | 0.04 | -0.62 |
| ann_daily_… | -0.04 | 0.98 | -0.18 | 0.23 | 1.00 | 0.79 | -0.18 |
| beta | 0.01 | 0.82 | -0.09 | 0.04 | 0.79 | 1.00 | 0.13 |
| cor | 0.67 | -0.17 | 0.70 | -0.62 | -0.18 | 0.13 | 1.00 |
Portfolios
By cluster: top 5 by Sharpe in each of 4 clusters (equal weight).
Top 20: top 20 stocks by Sharpe ratio overall (equal weight).
Portfolio by cluster (symbols)
CBOE, BRK-B, CB, ATO, ADP, CTRA, CF, DVN, CHRW, SCHW, CEG, FICO, CTAS, COST, COR, EQT, AMD, CRWD, DDOG, AMAT
Top 20 by Sharpe (symbols)
CEG, FICO, CTAS, COST, COR, ANET, AVGO, ACGL, AJG, ETN, BLDR, AXON, DECK, APO, AZO, AFL, CAH, BSX, BRO, APH
Validation: cumulative returns
Out-of-sample wealth paths (indexed to 1 at validation start) for the cluster portfolio, top-20 Sharpe basket, and S&P 500. Period: 2024-12-03 → 2026-06-05.
