pip install git+https://github.com/tensorflow/skflow import skflow classifier = skflow.TensorFlowDNNClassifier(hidden_units=[10, 20, 10], n_classes=2) classifier.fit(X_train, y_train) score = classifier.score(X_test, y_test) classifier.save('model') classifier = skflow.TensorFlowDNNClassifier.restore('model')