Note
Go to the end to download the full example code or to run this example in your browser via JupyterLite or Binder.
Risk Parity - Covariance shrinkage#
This tutorial shows how to incorporate covariance shrinkage in the
RiskBudgeting optimization.
Data#
We load the S&P 500 dataset composed of the daily prices of 20 assets from the S&P 500 Index composition starting from 1990-01-02 up to 2022-12-28:
from plotly.io import show
from sklearn.model_selection import train_test_split
from skfolio import Population, RiskMeasure
from skfolio.datasets import load_sp500_dataset
from skfolio.moments import ShrunkCovariance
from skfolio.optimization import RiskBudgeting
from skfolio.preprocessing import prices_to_returns
from skfolio.prior import EmpiricalPrior
prices = load_sp500_dataset()
X = prices_to_returns(prices)
X_train, X_test = train_test_split(X, test_size=0.33, shuffle=False)
Model#
We create a risk parity model by using ShrunkCovariance as
the covariance estimator then fit it on the training set:
model = RiskBudgeting(
risk_measure=RiskMeasure.VARIANCE,
prior_estimator=EmpiricalPrior(
covariance_estimator=ShrunkCovariance(shrinkage=0.9)
),
portfolio_params=dict(name="Risk Parity - Covariance Shrinkage"),
)
model.fit(X_train)
model.weights_
array([0.04774104, 0.04370383, 0.04503156, 0.04647703, 0.05284701,
0.0490748 , 0.04852963, 0.0537401 , 0.04539447, 0.05360773,
0.05178556, 0.0513795 , 0.0492718 , 0.05375803, 0.05112799,
0.05417665, 0.04754918, 0.0498816 , 0.05199232, 0.05293018])
To compare this model, we use a basic risk parity without covariance shrinkage:
bench = RiskBudgeting(
risk_measure=RiskMeasure.VARIANCE,
portfolio_params=dict(name="Risk Parity - Basic"),
)
bench.fit(X_train)
bench.weights_
array([0.04135436, 0.03210836, 0.03372654, 0.03784985, 0.06105106,
0.04432754, 0.04252248, 0.06593727, 0.03451718, 0.06469189,
0.05418594, 0.0520975 , 0.04535395, 0.06568141, 0.05104087,
0.0689447 , 0.04046515, 0.04667648, 0.05627082, 0.06119665])
Prediction#
We predict the model and the benchmark on the test set:
ptf_model_test = model.predict(X_test)
ptf_bench_test = bench.predict(X_test)
Analysis#
For improved analysis, it’s possible to load both predicted portfolios into a
Population:
population = Population([ptf_model_test, ptf_bench_test])
Let’s plot each portfolio cumulative returns:
fig = population.plot_cumulative_returns()
show(fig)
Finally, we print a full summary of both strategies evaluated on the test set:
population.summary()
Total running time of the script: (0 minutes 2.547 seconds)