import tensorflow.contrib.learn as skflow classifier = skflow.TensorFlowLinearClassifier(n_classes=2) classifier.fit(X_train, y_train, steps=1000) X_test = [[5], [6]] predictions = list(classifier.predict(X_test)) for i, prediction in enumerate(predictions): print('Sample {}: Predicted={}'.format(i+1, prediction))