Tensorflow | 读取csv文件

最新推荐文章于 2021-10-11 20:22:58 发布

茁壮小草

最新推荐文章于 2021-10-11 20:22:58 发布

阅读量1.9w

点赞数 6

分类专栏： Python 文章标签： tensorflow CSV DNN

本文链接：https://blog.csdn.net/xxzhangx/article/details/54667234

版权

如何将CSV数据读入到tensorflow中，这个问题困扰了我好几天，下面来说一种我现在用到的方法。

待有新的读取方法，本帖保持更新

方法一：
以一个案例来切入：

#加载包
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import tensorflow as tf
import numpy as np

# 数据集名称，数据集要放在你的工作目录下
IRIS_TRAINING = "iris_training.csv"
IRIS_TEST = "iris_test.csv"

# 数据集读取，训练集和测试集
training_set = tf.contrib.learn.datasets.base.load_csv_with_header(
    filename=IRIS_TRAINING,
    target_dtype=np.int,
    features_dtype=np.float32)
test_set = tf.contrib.learn.datasets.base.load_csv_with_header(
    filename=IRIS_TEST,
    target_dtype=np.int,
    features_dtype=np.float32)

# 特征
feature_columns = [tf.contrib.layers.real_valued_column("", dimension=4)]

# 构建DNN网络，3层，每层分别为10,20,10个节点
classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,