基于注意机制的LSTM股价预测模型

最新推荐文章于 2024-04-10 13:49:34 发布

经济工科生

最新推荐文章于 2024-04-10 13:49:34 发布

阅读量468

点赞数 1

文章标签： lstm tensorflow 深度学习算法大数据

本文链接：https://blog.csdn.net/weixin_49330017/article/details/128928609

版权

1、导入相关库

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import tushare as ts
import keras
from keras.layers import Dense, LSTM, Dropout,
from keras.callbacks import EarlyStopping, ModelCheckpoint
from keras.models import load_model 
import json
import os
import random
import tensorflow as tf
import keras.backend as K

2、设置随机数和早停法

#随机数种子设置
seed_num = 42
os.environ['PYTHONHASHSEED']=str(seed_num)
random.seed(seed_num)
np.random.seed(seed_num)
tf.random.set_seed(seed_num)
from tensorflow.compat.v1.keras import backend as K
session_conf = tf.compat.v1.ConfigProto(intra_op_parallelism_threads=1,
                             inter_op_parallelism_threads=1)
sess=tf.compat.v1.Session(graph=tf.compat.v1.get_default_graph(),config=session_conf)
K.set_session(sess)