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

Merge remote-tracking branch 'origin/main' into main

Browse Source
This commit is contained in:
Young
2020-11-25 12:31:45 +00:00
parent 47cf46f0c2 67e7fdf80b
commit 80fb24d8c3
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
qlib/data/ops.py
View File

@@ -18,9 +18,8 @@ try:
from ._libs.rolling import rolling_slope, rolling_rsquare, rolling_resi
from ._libs.expanding import expanding_slope, expanding_rsquare, expanding_resi
except ImportError as err:
print(err)
print("Do not import qlib package in the repository directory")
sys.exit(-1)
print("Do not import qlib package in the repository directory!")
raise
__all__ = (
"Ref",
@@ -865,6 +864,8 @@ class Skew(Rolling):
"""
def __init__(self, feature, N):
if N != 0 and N < 3:
raise ValueError("The rolling window size of Skewness operation should >= 3")
super(Skew, self).__init__(feature, N, "skew")
@@ -885,6 +886,8 @@ class Kurt(Rolling):
"""
def __init__(self, feature, N):
if N != 0 and N < 4:
raise ValueError("The rolling window size of Kurtosis operation should >= 5")
super(Kurt, self).__init__(feature, N, "kurt")
@@ -1268,7 +1271,7 @@ class WMA(Rolling):
def weighted_mean(x):
w = np.arange(len(x))
w /= w.sum()
w = w / w.sum()
return np.nanmean(w * x)
if self.N == 0: