dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2022-08-19 21:59:20 +09:00
parent f2ce5ada61
commit 5975584e86
1 changed files with 32 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
stock/util/Stock2Vector.py
View File

@@ -370,38 +370,38 @@ class Stock2Vector(HTS):
batch_X, batch_Y = [], []
for i in range(W_VECTOR_SIZE - 1, size):
X = np.zeros((H_VECTOR_SIZE, W_VECTOR_SIZE))
X[0] = macd[i: i - W_VECTOR_SIZE + 1]
X[1] = diff_avg27[i: i - W_VECTOR_SIZE + 1]
X[2] = diff_avg3_avg27[i: i - W_VECTOR_SIZE + 1]
X[3] = diff_avg3_avg54[i: i - W_VECTOR_SIZE + 1]
X[4] = diff_avg6_avg27[i: i - W_VECTOR_SIZE + 1]
X[5] = diff_avg6_avg54[i: i - W_VECTOR_SIZE + 1]
X[6] = diff_avg9_avg27[i: i - W_VECTOR_SIZE + 1]
X[7] = diff_avg9_avg54[i: i - W_VECTOR_SIZE + 1]
X[8] = diff_avg12_avg27[i: i - W_VECTOR_SIZE + 1]
X[9] = diff_avg12_avg54[i: i - W_VECTOR_SIZE + 1]
X[10] = diff_change_lead1[i: i - W_VECTOR_SIZE + 1]
X[11] = diff_open_lead1[i: i - W_VECTOR_SIZE + 1]
X[12] = diff_close_lead1[i: i - W_VECTOR_SIZE + 1]
X[13] = diff_high_lead1[i: i - W_VECTOR_SIZE + 1]
X[14] = diff_low_lead1[i: i - W_VECTOR_SIZE + 1]
X[15] = rsi[i: i - W_VECTOR_SIZE + 1]
X[16] = rsis[i: i - W_VECTOR_SIZE + 1]
X[17] = diff_avg54[i: i - W_VECTOR_SIZE + 1]
X[18] = diff_change_base[i: i - W_VECTOR_SIZE + 1]
X[19] = diff_base_lead1[i: i - W_VECTOR_SIZE + 1]
X[20] = diff_open_base[i: i - W_VECTOR_SIZE + 1]
X[21] = diff_close_base[i: i - W_VECTOR_SIZE + 1]
X[22] = diff_high_base[i: i - W_VECTOR_SIZE + 1]
X[23] = diff_low_base[i: i - W_VECTOR_SIZE + 1]
X[24] = abs_avg_1[i: i - W_VECTOR_SIZE + 1]
X[25] = abs_avg_2[i: i - W_VECTOR_SIZE + 1]
X[26] = abs_avg_3[i: i - W_VECTOR_SIZE + 1]
X[27] = abs_avg_4[i: i - W_VECTOR_SIZE + 1]
X[28] = abs_avg_5[i: i - W_VECTOR_SIZE + 1]
X[29] = diff_upper_lower[i: i - W_VECTOR_SIZE + 1]
X[30] = diff_open_lower[i: i - W_VECTOR_SIZE + 1]
X[31] = diff_close_upper[i: i - W_VECTOR_SIZE + 1]
X[0] = macd[i - W_VECTOR_SIZE + 1: i + 1]
X[1] = diff_avg27[i - W_VECTOR_SIZE + 1: i + 1]
X[2] = diff_avg3_avg27[i - W_VECTOR_SIZE + 1: i + 1]
X[3] = diff_avg3_avg54[i - W_VECTOR_SIZE + 1: i + 1]
X[4] = diff_avg6_avg27[i - W_VECTOR_SIZE + 1: i + 1]
X[5] = diff_avg6_avg54[i - W_VECTOR_SIZE + 1: i + 1]
X[6] = diff_avg9_avg27[i - W_VECTOR_SIZE + 1: i + 1]
X[7] = diff_avg9_avg54[i - W_VECTOR_SIZE + 1: i + 1]
X[8] = diff_avg12_avg27[i - W_VECTOR_SIZE + 1: i + 1]
X[9] = diff_avg12_avg54[i - W_VECTOR_SIZE + 1: i + 1]
X[10] = diff_change_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[11] = diff_open_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[12] = diff_close_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[13] = diff_high_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[14] = diff_low_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[15] = rsi[i - W_VECTOR_SIZE + 1: i + 1]
X[16] = rsis[i - W_VECTOR_SIZE + 1: i + 1]
X[17] = diff_avg54[i - W_VECTOR_SIZE + 1: i + 1]
X[18] = diff_change_base[i - W_VECTOR_SIZE + 1: i + 1]
X[19] = diff_base_lead1[i - W_VECTOR_SIZE + 1: i + 1]
X[20] = diff_open_base[i - W_VECTOR_SIZE + 1: i + 1]
X[21] = diff_close_base[i - W_VECTOR_SIZE + 1: i + 1]
X[22] = diff_high_base[i - W_VECTOR_SIZE + 1: i + 1]
X[23] = diff_low_base[i - W_VECTOR_SIZE + 1: i + 1]
X[24] = abs_avg_1[i - W_VECTOR_SIZE + 1: i + 1]
X[25] = abs_avg_2[i - W_VECTOR_SIZE + 1: i + 1]
X[26] = abs_avg_3[i - W_VECTOR_SIZE + 1: i + 1]
X[27] = abs_avg_4[i - W_VECTOR_SIZE + 1: i + 1]
X[28] = abs_avg_5[i - W_VECTOR_SIZE + 1: i + 1]
X[29] = diff_upper_lower[i - W_VECTOR_SIZE + 1: i + 1]
X[30] = diff_open_lower[i - W_VECTOR_SIZE + 1: i + 1]
X[31] = diff_close_upper[i - W_VECTOR_SIZE + 1: i + 1]
batch_X.append(X)
batch_Y.append(label[i])