#加载包
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import tensorflow as tf
import numpy as np
# 数据集名称,数据集要放在你的工作目录下
IRIS_TRAINING = "d:/mhuanjing.csv"
IRIS_TEST = "d:/mhuanjing.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=18)]
# 构建DNN网络,3层,每层分别为10,20,10个节点
classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,hidden_units=[100, 200, 100],n_classes=7,model
python3 tensorflow
最新推荐文章于 2024-06-08 14:37:14 发布