Skip to content

Instantly share code, notes, and snippets.

@ilblackdragon
Last active August 27, 2016 22:00
Show Gist options
  • Save ilblackdragon/531ab1689c8077c5c684 to your computer and use it in GitHub Desktop.
Save ilblackdragon/531ab1689c8077c5c684 to your computer and use it in GitHub Desktop.
>>> classifier = learn.DNNClassifier(hidden_units=[10, 20, 10],
... n_classes=2,
... feature_columns=learn.infer_real_valued_columns_from_input(X_train),
... optimizer=tf.train.GradientDescentOptimizer(learning_rate=0.05))
>>> classifier.fit(X_train, y_train, batch_size=128, steps=500)
>>> score = accuracy_score(classifier.predict(X_test), y_test)
>>> print("Accuracy: %f" % score)
Accuracy: 0.67597765363
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment