liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-03 11:00:57 +08:00
Code Activity

Resolve https://github.com/microsoft/qlib/pull/280\#discussion_r589169769

Browse Source
This commit is contained in:
Charles Young
2021-03-08 19:02:40 +08:00
parent fc89fec46d
commit 2f9af1af8f
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
qlib/portfolio/optimizer/enhanced_indexing.py
View File

@@ -70,7 +70,7 @@ class EnhancedIndexingOptimizer(BaseOptimizer):
def __call__(
self,
u: np.ndarray,
u: Union[np.ndarray, pd.Series],
F: np.ndarray,
covB: np.ndarray,
varU: np.ndarray,
@@ -80,7 +80,7 @@ class EnhancedIndexingOptimizer(BaseOptimizer):
) -> Union[np.ndarray, pd.Series]:
"""
Args:
u (np.ndarray): expected returns (a.k.a., alpha)
u (np.ndarray or pd.Series): expected returns (a.k.a., alpha)
F, covB, varU (np.ndarray): see StructuredCovEstimator
w0 (np.ndarray): initial weights (for turnover control)
w_bench (np.ndarray): benchmark weights
@@ -91,6 +91,10 @@ class EnhancedIndexingOptimizer(BaseOptimizer):
"""
assert inds_onehot is not None or self.inds_dev is None, "Industry onehot vector is required."
# transform dataframe into array
if isinstance(u, pd.Series):
u = u.values
# scale alpha to match volatility
if self.scale_alpha:
u = u / u.std()