使用 TensorFlow 的起始步骤

使用 TensorFlow 的起始步骤

TensorFlow 工具包的层次结构。Estimators 位于顶级。
import tensorflow as tf

# Set up a linear classifier.
classifier = tf.estimator.LinearClassifier()

# Train the model on some example data.
classifier.train(input_fn=train_input_fn, steps=2000)

# Use it to predict.
predictions = classifier.predict(input_fn=predict_input_fn)