TensorflowでアヤメのNNのテスト

ここを参考に
https://www.tensorflow.org/get_started/tflearn

import numpy as np
import tensorflow as tf
from sklearn import datasets

iris = datasets.load_iris()
data = iris.data
target = iris.target
def train():
    return (tf.constant(data),tf.constant(target))
def test():
    return (tf.constant(data),tf.constant(target))

feature_columns = [tf.contrib.layers.real_valued_column("", dimension=4)] # dimension->features' dimension
classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
                                              hidden_units=[10, 20, 10],
                                              n_classes=3,
                                              model_dir="/tmp/temp_model")
classifier.fit(input_fn=train, steps=2000)
accuracy_score = classifier.evaluate(input_fn=test,
                                       steps=1)["accuracy"]