Skip to content

Instantly share code, notes, and snippets.

@hsmnzaydn

hsmnzaydn/Train

Last active April 1, 2018 17:18
Show Gist options
  • Save hsmnzaydn/6a9585248571d417a4c6cb81a274f98c to your computer and use it in GitHub Desktop.
Save hsmnzaydn/6a9585248571d417a4c6cb81a274f98c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Train
X_train, X_test, y_train, y_test = train_test_split(X, Y)
estimators = []
estimator = KerasRegressor(build_fn=model, epochs=20, batch_size=50, verbose=2)
export_model(tf.train.Saver(), estimator, ["input name"], "output name","test")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment