第三方库:
tensorflow 2.1
tensorflow-cpu 2.1
numpy 1.18.1
matplotlib 3.2.0
pandas 1.0.1
使用教程:
1、如果想重新训练模型,则只需要以下几步:
a、读取CSV文件,(文件内容与文件格式一样)
b、将数据传入trainSet函数,注释掉主函数中的其他函数,此时trainSet函数就会训练模型,保存模型参数,并在本文件目录下生成一个文件夹model_save。
2、如果想测试模型,则只需要以下几步:
a、读取CSV文件,(文件内容与文件格式一样)
b、调用testSet(),注释掉主函数中的其他函数,
3、如果想预测,则调用prediction()函数,注释掉主函数中的其他函数。
-
Notifications
You must be signed in to change notification settings - Fork 0
weekstudy/stock_prediction-with-LSTM
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published