Skip to content

Instantly share code, notes, and snippets.

@securetorobert

securetorobert/tf_l1_reg_prediction.py

Created July 15, 2018 21:50
Show Gist options
  • Save securetorobert/dc56e3cc05c1dd9a7f0774289ed060b1 to your computer and use it in GitHub Desktop.
Save securetorobert/dc56e3cc05c1dd9a7f0774289ed060b1 to your computer and use it in GitHub Desktop.
Download ZIP
Make a prediction on l1 reg model in TensorFlow
Raw
tf_l1_reg_prediction.py
preds = estimator.predict(input_fn=lambda: input_fn(X_test, y_test))
predictions = np.array([item['predictions'][0] for item in preds])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment