利用Tensorflow里的LSTM对北京PM2.5数据集Beijing PM2.5 Data Data Set进行预测

最新推荐文章于 2026-06-09 21:47:22 发布

原创

最新推荐文章于 2026-06-09 21:47:22 发布 · 1.3w 阅读

收录于

当前文章被以下社区和专栏收录：

该博客介绍了如何运用Tensorflow的LSTM模型对北京2010-2014年PM2.5数据集进行预测。首先对数据进行了标准化处理，然后引用了两篇有关LSTM的资源进行学习。尽管训练效果不理想，测试集的均方误差较高，但博主分享了完整的代码实现。

课程作业。

数据集：Beijing PM2.5 Data Data Set

这个数据集里有北京2010-2014的天气数据。每小时一条，适合用于回归模型。

这里我对数据进行了标准化处理

在学习LSTM的时候这两篇文章给了我很大帮助：

http://karpathy.github.io/2015/05/21/rnn-effectiveness/

https://www.zybuluo.com/hanbingtao/note/581764

直接上代码，代码很大程度上参考了某篇博文，在这里暂时找不到原代码链接了，以后有机会再补上。

# -*- coding: utf-8 -*-
"""
Created on Wed Jun  6 10:56:52 2018

@author: Administrator
"""

import tensorflow as tf
import pandas as pd
import numpy as np
import csv

tf.reset_default_graph()

rnn_unit = 50       #the amount of hidden lstm units
batch_size = 72     #the amount of data trained every time
in

标签