ここを参考に
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"]