Note
Go to the end to download the full example code or to run this example in your browser via JupyterLite.
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.04774404, 0.04370342, 0.04503321, 0.04647533, 0.05284521,
0.04907493, 0.04852809, 0.05373997, 0.04539475, 0.05360793,
0.05178586, 0.0513793 , 0.04927117, 0.05375824, 0.05112857,
0.05417612, 0.0475513 , 0.04988039, 0.05199224, 0.05292994])
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.04135318, 0.03210824, 0.0337272 , 0.03785076, 0.06105329,
0.04432756, 0.04252214, 0.06593567, 0.03451845, 0.06469326,
0.05418886, 0.05209518, 0.04535298, 0.0656811 , 0.0510388 ,
0.06894497, 0.04046537, 0.04667777, 0.05626949, 0.06119573])
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.740 seconds)