1. 程式人生 > >數據遊戲Round1:預測5月15號招商銀行的股價

數據遊戲Round1:預測5月15號招商銀行的股價

ttr 做了 數學 知識 所有 增加 預測 select dex

  前陣子報名參加了一個數據比賽,題目是預測5月15號(星期三)招商銀行的股價,截止時間是在5月12號(星期天)。在本次預測中,我用到的是嶺回歸。

嶺回歸

  嶺回歸是回歸的一種,它解決回歸中重大疑難問題:排除多重共線性,進行變量的選擇,在存在共線性問題和病態數據偏多的研究中有較大的實用價值。按照度娘百科的解釋:嶺回歸是一種專用於共線性數據分析的有偏估計回歸方法,實質上是一種改良的最小二乘估計法,通過放棄最小二乘法的無偏性,以損失部分信息、降低精度為代價獲得回歸系數更為符合實際、更可靠的回歸方法,對病態數據的擬合要強於最小二乘法。

  多重共線性指線性回歸模型中的解釋變量之間由於存在精確相關關系或高度相關關系而使模型估計失真或難以估計準確。

  由於嶺回歸涉及的數學知識比較復雜,且文章篇幅有限,這裏不過多展開。

數據獲取

  本次數據是通過 Tushare 的 get_hist_data()獲取的。Tushare是一個免費、開源的python財經數據接口包。python安裝tushare直接通過
pip install tushare 即可安裝。

import tushare as ts
data = ts.get_hist_data(600848)

  運行之後可以查看它的前後幾行數據,按照tushare官方的說明,get_hist_data()只能獲取近3年的日線數據,而他的返回值的說明是這樣的:

〖date:日期;open:開盤價;high:最高價;close:收盤價;low:最低價;volume:成交量;price_change:價格變動;p_change:漲跌幅;ma5:5日均價;ma10:10日均價;ma20:20日均價;v_ma5:5日均量;v_ma10:10日均量;v_ma20:20日均量〗

  均價的意思大概就是股票n天的成交價格或指數的平均值。均量則跟成交量有關。至於其他的返回值,應該是一下子就能明白的吧。在獲得數據之後,我們查看一下描述性統計,通過 data.describe() 查看是否存在什麽異常值或者缺失值。

   技術分享圖片

  這樣看來似乎除了由於周末以及節假日不開盤導致的當天的數據缺失以外,並沒有其他的缺失和異常。但是這裏我們不考慮節假日的缺失值。

數據預處理

  由於獲取的數據是按日期降序排序,但本次預測跟時間序列有關,因此我們需要把順序轉一下,讓它按照日期升序排序。

data1 = data[::-1]

  處理完順序之後,我們要做一下特征值的選擇。由於 volume 以及均量的值很大,如果不進行處理的話,很可能對整體的預測造成不良影響。由於時間有限,而且考慮到運算的復雜度,這裏我沒有對這些特征進行處理,而是直接將它們去掉了。至於均價,我是按照自己的理解,和10日均價、20日均價相比,5日均價的範圍沒那麽大,對近期的預測會比另外兩個要好,因此保留5日均價。接著,我用 sklearn.model_selection 的 cross_val_score,分別查看除〖‘open‘, ‘close‘, ‘high‘, ‘low‘, ‘ma5‘〗以外的其他剩余屬性對預測值的影響。發現 ‘p_change’、‘price_change‘ 這兩個屬性對預測結果的影響不大,為了節省內存,增加運算速度,提高預測的準確性,也直接把它們去掉了。完了之後,查看前後三行數據。

data1 = data1[[open,high,low,ma5,close]]
data1.head(3), data1.tail(3)

        技術分享圖片

建模預測

  由於提交截止日期是周日,預測的是周三,因此需要先對周一周二的信息進行預測。在這裏我突然想到一個問題,是用前一天的所有數據來訓練模型以預測當天的 close 比較準確,還是用當天除了 close 以外的其他數據來訓練模型以訓練當天的 close 比較準呢?為了驗證這個問題,我分別對這兩種方法做了實驗。

  為了減少代碼量,定義了一個函數用以評估模型的錯誤率。

def get_score(X_train, y_train):
    ridge_score = np.sqrt(-cross_val_score(ridge, X_train, y_train, cv=10, scoring=neg_mean_squared_error))
    return np.mean(ridge_score)

 (1)、用前一天的所有數據來當訓練集

y_train = data1[close].values[1:]
X_train = data1.values[:-1]
score = get_score(X_train, y_train)

  輸出結果大約為0.469,這個錯誤率就比較大了,不太合理,更何況還要預測其他特征值作為測試數據。

(2)、用當天除了 close 以外的其他數據來當訓練集

data2 = data1[:]
y_train = data2.pop(close).values
X_train = data2.values
score = get_score(X_train, y_train)

  輸出結果大約為0.183,跟第一個相比簡直好多了。所以,就決定是你了!

  接下來建模並把模型保存下來:

y_train = data1[close]
X_train = data1[[open, high, low, ma5]]
close_model = ridge.fit(X_train, y_train)
joblib.dump(ridge, close_model.m)

  在預測之前呢,我們先拿訓練集的後8組數據做一下測試,做個圖看看:

scores = []
for x in X_train[-8:]:
    score = close_model.predict(np.array(x).reshape(1, -1))
    scores.append(score)
x = np.arange(8)
fig, axes = plt.subplots(1, 1, figsize=(13, 6))
axes.plot(scores)
axes.plot(y_train[-8:])
plt.xticks(x, data1.index[-8:].values, size=13, rotation=0)

  技術分享圖片

  看到這樣子我還是相對比較放心的,不過,這個模型的訓練值除了“close”以外的屬性都是已知的,要預測三天後的還得預測前兩天的測試值。

def get_model(s):
    y_train = data1[s].values[1:]
    X_train = data1.values[:-1]
    model = ridge.fit(X_train, y_train)
    return model
def get_results(X_test):
    attrs = [open, high, low, ma5]
    results = []
    for attr in attrs:
        result = get_model(attr).predict(X_test)
        results.append(result)
    return results

  接下來預測三天的股價:

X_test = data1[-1:].values
for i in range(3):
    results = get_results(X_test)
    close = close_model.predict(np.array(results).reshape(1, -1))
    print(close)
    results.append(close)
    X_test = np.array(results).reshape(1, -1)
print("5月15日招商銀行關盤時的股價為:" + str(round(close[0], 2)))
[33.46370029]
[33.4532047]
[33.43836721]
5月15日招商銀行關盤時的股價為:33.44

總結

  雖然預測結果是這樣子,但感覺這樣預測似乎很菜啊。畢竟預測的每個值都會有偏差,多個偏差累加起來就很多了,這讓我有點害怕。不知道存不存在不預測其他值直接預測close的方法,或者說直接預測5月15號的而不用先預測13、14號的方法。雖然我知道有種算法是時間序列算法,但不是很懂。希望哪位大神看了能給我一些建議,指點迷津。

  對於一個自學數據分析的在校學生,苦於沒有項目經驗,正好趕上這次的【數據遊戲】,能利用此次機會操作一波真的很不錯。

數據遊戲Round1:預測5月15號招商銀行的股價