Skip to content

Instantly share code, notes, and snippets.

@Ben-Epstein

Ben-Epstein/Deploy Model.py

Last active November 5, 2020 15:41
Show Gist options
  • Save Ben-Epstein/e47dca85c6a065238308a24a9ef77535 to your computer and use it in GitHub Desktop.
Save Ben-Epstein/e47dca85c6a065238308a24a9ef77535 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Deploy Model.py
schema = splice.getConnection().getSchema()
run_id = mlflow.current_run_id()
splice.dropTableIfExists(schema, 'iris_model')
jid = mlflow.deploy_db(schema, 'iris_model', run_id, create_model_table=True, df=df.select(cols[:-1]), primary_key={'MOMENT_ID':'INT'}, classes=list(data.target_names))
mlflow.watch_job(jid)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment