Skip to content

Violettttee/Pytorch-lstm-attention

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-cnn-lstm-forecast

基于pytorch搭建多特征CNN-LSTM时间序列预测,运行main.py即可,若要保存模型需手动添加相应代码……

Datasets

Kaggle上的房价预测

Attention

在模型上添加了attention机制,具体拟合效果可见图

原数据按照月度聚合

alt text

原数据按照季度聚合

alt text

单分类上的拟合(在bedrooms)

alt text

多分类上的拟合(未加入attention前)10w次iter

alt text

多分类上的拟合(加入attention后)10w次iter

alt text

About

加了attention机制的多特征lstm预测模型

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages