[deeplearning-017]一个最简的tensorflow graph execution示例

最新推荐文章于 2023-11-19 20:39:01 发布

未济2019

最新推荐文章于 2023-11-19 20:39:01 发布

阅读量847

点赞数

本文链接：https://blog.csdn.net/lizhe_dashuju/article/details/82080882

版权

本文提供了一个简单的 TensorFlow 图执行示例，通过参考 TensorFlow 官方的 demo，展示如何构建和执行计算图。

摘要由CSDN通过智能技术生成

1.主要参考tf的demo

2.源码



import tensorflow as tf
import pandas as pd

CSV_COLUMN_NAMES = ['SepalLength', 'SepalWidth', 'PetalLength', 'PetalWidth', 'Species']
SPECIES = ['Setosa', 'Versicolor', 'Virginica']

#训练样本和测试样本
#"http://download.tensorflow.org/data/iris_training.csv"
#"http://download.tensorflow.org/data/iris_test.csv"

tf.logging.set_verbosity(tf.logging.INFO)

train = pd.read_csv('./iris_training.csv', names=CSV_COLUMN_NAMES, header=0)
train_y = train.pop('Species')
train_x = train

def input_fn_1(features, labels, batch_size):
    dataset = tf.data.Dataset.from_tensor_slices((dict(features), labels))
    dataset = dataset.shuffle(1000).repeat().batch(batch_size)
    return dataset

def input_fn_2(features, labels, batch_size):
    features = dict(features)
    if labels is None: