1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-15 17:06:44 +00:00

add files

This commit is contained in:
2020-04-29 18:57:37 +08:00
parent c01add193e
commit c11466c958
2 changed files with 11 additions and 13 deletions

View File

@@ -80,6 +80,7 @@ def train(config, train_X, train_Y, valid_X, valid_Y):
def predict(config, test_X):
config.dropout_rate = 0.1
tf.reset_default_graph()
with tf.variable_scope("stock_predict", reuse=tf.AUTO_REUSE):
model = Model(config)