def getTrainData_1(symbol, startDate, endDate, periods=14, remove_head=19):
dates = pd.date_range(startDate, endDate)
df = util.loadPriceData([symbol], dates)
util.fill_missing_values(df)
# skip periods day latest (14 days) that predict a price is new high or not
df_sliced = df.ix[0:len(df) - periods]
price_close = pd.DataFrame(df_sliced[symbol])
set = pd.DataFrame(df_sliced['SET'])
rsi = ind.rsi(price_close) / 100 # normalize
sr = ind.create_dataframe_SR(df_sliced, [symbol])
myRatio = ind.get_myRatio(price_close)
daily = ind.daily_returns(price_close) * 100 # normalize
_, _, macd = ind.average_convergence(price_close)
ohlcv = util.load_OHLCV(symbol, dates)
percent_KD = ind.percent_KD(ohlcv) / 100 # normalize
c2o = ind.daily_returns_2(ohlcv) * 100 # normalize
volume = util.loadVolumeData([symbol], dates)
#skip periods day latest
volume_sliced = volume.ix[0:len(volume) - periods]
assert len(volume_sliced) == len(df_sliced)
obv = ind.OBV(volume_sliced, df_sliced)
obv_rsi = ind.rsi(obv) / 100 # calcuate momentum with rsi and normalize
set_rsi = ind.rsi(set) / 100 # calcuate momentum with rsi and normalize
# Join data frame
# rename column
rsi.rename(columns={symbol: 'RSI'}, inplace=True)
#sr.rename(columns={symbol:'SR'},inplace=True)
myRatio.rename(columns={symbol: 'MY'}, inplace=True)
daily.rename(columns={symbol: 'DAILY'}, inplace=True)
macd.rename(columns={symbol: 'MACD'}, inplace=True)
obv_rsi.rename(columns={symbol: 'OBV_RSI'}, inplace=True)
set_rsi.rename(columns={'SET': 'SET_RSI'}, inplace=True)
Xtrain = pd.DataFrame(index=df_sliced.index)
Xtrain = Xtrain.join(rsi['RSI'])
Xtrain = Xtrain.join(percent_KD['%K'])
#Xtrain = Xtrain.join(sr['SR'])
Xtrain = Xtrain.join(myRatio['MY'])
Xtrain = Xtrain.join(daily['DAILY'])
Xtrain = Xtrain.join(macd['MACD'])
Xtrain = Xtrain.join(c2o)
Xtrain = Xtrain.join(obv_rsi['OBV_RSI'])
Xtrain = Xtrain.join(set_rsi)
upTrend = ind.isUpTrend(df, symbol, periods=periods)
Ydigit = 1 * upTrend[
symbol] # multiply 1 : True is converted to 1 (up trend) , False becomes 0
# skip at head row, avoid NaN values
Xtrain = Xtrain.ix[remove_head:]
Ydigit = Ydigit.ix[remove_head:]
price_close = price_close.ix[remove_head:]
return Xtrain, Ydigit, price_close
def getTrainData_4(symbol, startDate, endDate, periods=14, remove_head=19):
dates = pd.date_range(startDate, endDate)
# day periods that predict a price is new high or not
df = util.loadPriceData([symbol], dates)
# skip periods day latest
df_sliced = df.ix[0:len(df) - periods]
price_close = pd.DataFrame(df_sliced[symbol])
set = pd.DataFrame(df_sliced['SET'])
bbands = ind.get_BBANDS(price_close, symbol)
ema26, ema12, macd = ind.average_convergence(price_close)
rsi = ind.rsi(price_close)
daily = ind.daily_returns(price_close)
ohlcv = util.load_OHLCV(symbol, dates)
percent_KD = ind.percent_KD(ohlcv)
volume = util.loadVolumeData(symbol, dates)
#skip periods day latest
volume_sliced = volume.ix[0:len(volume) - periods]
assert len(volume_sliced) == len(price_close)
volume_sliced = pd.DataFrame(volume_sliced[symbol])
obv = ind.OBV(volume_sliced, price_close)
# Join data frame
# rename column
price_close.rename(columns={symbol: 'CLOSE'}, inplace=True)
ema26.rename(columns={symbol: 'EMA26'}, inplace=True)
ema12.rename(columns={symbol: 'EMA12'}, inplace=True)
daily.rename(columns={symbol: 'DAILY'}, inplace=True)
rsi.rename(columns={symbol: 'RSI'}, inplace=True)
obv.rename(columns={symbol: 'OBV'}, inplace=True)
Xtrain = price_close
Xtrain = Xtrain.join(bbands['UPPER'])
Xtrain = Xtrain.join(bbands['LOWER'])
Xtrain = Xtrain.join(ema26)
Xtrain = Xtrain.join(ema12)
Xtrain = Xtrain.join(rsi)
Xtrain = Xtrain.join(percent_KD['%K'])
Xtrain = Xtrain.join(obv)
Xtrain = Xtrain.join(set)
upTrend = ind.isUpTrend(df, symbol, periods=periods)
Ydigit = 1 * upTrend[
symbol] # multiply 1 : True is converted to 1 (up trend) , False becomes 0
assert len(Xtrain) == len(Ydigit)
# skip at head row, avoid NaN values
Xtrain = Xtrain.ix[remove_head:]
Ydigit = Ydigit.ix[remove_head:]
Xtrain.fillna(0, inplace=True) # protected NaN value
return Xtrain, Ydigit
def plotDailyHist(df, symbol, title="Selected data"):
"""Plot the desired columns over index values in the given range."""
df_plot = ind.daily_returns(df)[symbol]
"""Plot stock prices with a custom title and meaningful axis labels."""
df_plot.hist(bins=20)
ax = plt.axvline(df_plot.mean(),
color='w',
linestyle='dashed',
linewidth=2)
std = df_plot.std()
plt.axvline(std, color='r', linestyle='dashed', linewidth=2)
plt.axvline(-std, color='r', linestyle='dashed', linewidth=2)
#ax.set_xlabel("Daily returns")
#ax.set_ylabel("Frequency")
plt.show()
def getTrainData_3(symbol, startDate, endDate, remove_head=15):
dates = pd.date_range(startDate, endDate)
df = util.loadPriceData([symbol], dates) #1
close = df.ix[0:len(df) - 1] # skip tail
assert len(df) == len(close) + 1
ema5 = ind.ema(close, periods=5) #2
sma5 = ind.ema(close, periods=5) #3
ema15 = ind.ema(close, periods=5) #4
sma15 = ind.ema(close, periods=5) #5
bb1p = ind.get_BBANDS(close, symbol, periods=14, mul=1) #6, #7
bb2p = ind.get_BBANDS(close, symbol, periods=14, mul=2) #8, #9
_, _, macd = ind.average_convergence(close) #10
signal_macd = ind.signal_MACD(macd) #11
rsi = ind.rsi(close) #12
ohlcv = util.load_OHLCV(symbol, dates)
percent_KD = ind.percent_KD(ohlcv) #14, 15
# rename columns
close.rename(columns={symbol: 'CLOSE'}, inplace=True)
ema5.rename(columns={symbol: 'EMA5'}, inplace=True)
sma5.rename(columns={symbol: 'SMA5'}, inplace=True)
#ema15.rename(columns={symbol:'EMA15'},inplace=True)
#sma15.rename(columns={symbol:'SMA15'},inplace=True)
bb1p.rename(columns={'LOWER': 'LOWER_BB1P'}, inplace=True)
bb1p.rename(columns={'UPPER': 'UPPER_BB1P'}, inplace=True)
#bb2p.rename(columns={'LOWER':'LOWER_BB2P'},inplace=True)
#bb2p.rename(columns={'UPPER':'UPPER_BB2P'},inplace=True)
macd.rename(columns={symbol: 'MACD'}, inplace=True)
#signal_macd.rename(columns={symbol:'SG_MACD'},inplace=True)
rsi.rename(columns={symbol: 'RSI'}, inplace=True)
Xtrain = pd.DataFrame(index=close.index)
Xtrain = Xtrain.join(close['CLOSE']) # 1
Xtrain = Xtrain.join(ema5['EMA5']) # 2
Xtrain = Xtrain.join(sma5['SMA5']) # 3
#Xtrain = Xtrain.join(ema15['EMA15']) # 4
#Xtrain = Xtrain.join(sma15['SMA15']) # 5
Xtrain = Xtrain.join(bb1p['LOWER_BB1P']) # 6
Xtrain = Xtrain.join(bb1p['UPPER_BB1P']) # 7
#Xtrain = Xtrain.join(bb2p['LOWER_BB2P']) # 8
#Xtrain = Xtrain.join(bb2p['UPPER_BB2P']) # 9
Xtrain = Xtrain.join(macd['MACD']) # 10
#Xtrain = Xtrain.join(signal_macd['SG_MACD']) # 11
Xtrain = Xtrain.join(rsi['RSI']) # 12
Xtrain = Xtrain.join(percent_KD) # 13
df = df[symbol] # skip SET
daily = ind.daily_returns(df)
daily = daily.shift(-1) # predict tommorow
daily.dropna(0, inplace=True) # drop NaN in last row
Ytrain = 1 * (daily > 0.0) # if positive it converted 1, or 0 in negative
assert len(Ytrain) == len(Xtrain)
# skip at head row, avoid NaN values
Xtrain = Xtrain[remove_head:]
Ytrain = Ytrain[remove_head:]
# normalize
normalizer = preprocessing.Normalizer().fit(Xtrain)
Xnorm = normalizer.transform(Xtrain)
Xnorm = pd.DataFrame(Xnorm, columns=Xtrain.columns)
return Xnorm, Ytrain
