用深度神经网络对boston housing data进行回归预测的程序--tensorflow
2017-05-08 22:12
337 查看
"""DNNRegressor with custom input_fn for Housing dataset.""" from __future__ import absolute_import from __future__ import division from __future__ import print_function import itertools import pandas as pd import tensorflow as tf tf.logging.set_verbosity(tf.logging.INFO) COLUMNS = ["crim", "zn", "indus", "nox", "rm", "age", "dis", "tax", "ptratio", "medv"] FEATURES = ["crim", "zn", "indus", "nox", "rm", "age", "dis", "tax", "ptratio"] LABEL = "medv" def input_fn(data_set): feature_cols = {k: tf.constant(data_set[k].values) for k in FEATURES} labels = tf.constant(data_set[LABEL].values) return feature_cols, labels def main(unused_argv): # Load datasets training_set = pd.read_csv("boston_train.csv", skipinitialspace=True, skiprows=1, names=COLUMNS) test_set = pd.read_csv("boston_test.csv", skipinitialspace=True, skiprows=1, names=COLUMNS) # Set of 6 examples for which to predict median house values prediction_set = pd.read_csv("boston_predict.csv", skipinitialspace=True, skiprows=1, names=COLUMNS) # Feature cols feature_cols = [tf.contrib.layers.real_valued_column(k) for k in FEATURES] # Build 2 layer fully connected DNN with 10, 10 units respectively. regressor = tf.contrib.learn.DNNRegressor(feature_columns=feature_cols, hidden_units=[10, 10], model_dir="/tmp/boston_model") # Fit regressor.fit(input_fn=lambda: input_fn(training_set), steps=5000) # Score accuracy ev = regressor.evaluate(input_fn=lambda: input_fn(test_set), steps=1) loss_score = ev["loss"] print("Loss: {0:f}".format(loss_score)) # Print out predictions y = regressor.predict(input_fn=lambda: input_fn(prediction_set)) # .predict() returns an iterator; convert to a list and print predictions predictions = list(itertools.islice(y, 6)) print("Predictions: {}".format(str(predictions))) if __name__ == "__main__": tf.app.run()
相关文章推荐
- 利用回归树对Boston房价进行预测,并对结果进行评估
- 用深度神经网络对Iris数据集进行分类的程序--tensorflow
- K近邻回归模型对Boston房价进行预测,同时对性能进行评估(1.使用普通的算术平均法2.考虑距离差异进行加权平均)
- 分别用普通线性回归、岭回归、Lasso回归对boston房价进行预测,比较效果
- 集成模型(回归)对Boston数据进行训练并对数据做出预测
- Tensorflow案例分析(1)—— 使用逻辑回归进行泰坦尼克号预测
- tensorflow下对MNIST数据集进行识别的程序代码
- 用卷积神经网络对mnist进行数字识别程序(tensorflow)
- 用sklearn和tensorflow做boston房价的回归计算的比较(2)--卷积神经网路CNN
- 用sklearn和tensorflow做boston房价的回归计算的比较(3)--RNN之递归神经网路LSTM
- 使用libsvm进行回归预测
- Tensorflow:深度神经网络DNN预测波士顿房价(boston house price)【一】
- tensorflow保存网络参数 使用训练好的网络参数进行数据的预测
- 极限学习机︱R语言快速深度学习进行回归预测
- 利用回归方程进行预测
- 【大数据部落】R语言逻辑回归logistic regression对用户收入进行预测
- 看深度学习框架排名第一的TensorFlow如何进行时序预测!
- 用sklearn和tensorflow做boston房价的回归计算的比较(1)--经典的sklearn集成模型
- R语言快速深度学习进行回归预测(转)
- 简单数据预测—使用Python训练回归模型并进行预测(转自蓝鲸网站分析博客)