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.04774397, 0.04370323, 0.0450325 , 0.04647489, 0.0528455 ,
0.04907489, 0.04852859, 0.05373967, 0.04539503, 0.05360706,
0.0517855 , 0.05138157, 0.04927105, 0.05375764, 0.0511287 ,
0.05417589, 0.04755072, 0.04988147, 0.05199201, 0.05293012])
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.04135311, 0.03210833, 0.0337259 , 0.03785133, 0.06105386,
0.04432976, 0.04252257, 0.06593182, 0.03451821, 0.06469574,
0.05419076, 0.0520937 , 0.04535439, 0.06567909, 0.05104142,
0.06894221, 0.04046537, 0.04667614, 0.0562728 , 0.06119348])
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 0.576 seconds)