Skip to content

Instantly share code, notes, and snippets.

@aymericdelab
Created October 17, 2019 14:40
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aymericdelab/fa756483a1328a1817d8b5c5c0e703e0 to your computer and use it in GitHub Desktop.
Save aymericdelab/fa756483a1328a1817d8b5c5c0e703e0 to your computer and use it in GitHub Desktop.
create your estimator and submit it to Azure for training
from azureml.train.dnn import TensorFlow
datastore = ws.get_default_datastore()
script_params = {
'--data-folder': datastore.as_mount(),
'--batch-size': 32,
'--learning-rate': 0.001,
'--prefix': 'founder-classifier',
'--steps': 1000
}
estimator = TensorFlow(source_directory=script_folder,
script_params=script_params,
compute_target=compute_target,
entry_script='azure_entry_point.py',
pip_packages=None)
run = exp.submit(estimator)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment