-
Notifications
You must be signed in to change notification settings - Fork 1.2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added export to MLflow pyfunc model format #1192
Merged
Merged
Changes from all commits
Commits
Show all changes
39 commits
Select commit
Hold shift + click to select a range
c664a43
Refactored callbacks
tgaddair 081b465
Added callbacks to load
tgaddair 562cb7c
Made Comet into callback
tgaddair 8926a6e
Converted API
tgaddair 99c26b2
Fixed hyperopt
tgaddair f1010b1
Fixed remainder
tgaddair 677a59d
Removed non-visualize
tgaddair 9c6cd59
visualize_callbacks
tgaddair 13330f1
Removed Mlflow for now
tgaddair cd0bb60
Fixed tests
tgaddair 690f486
Fixed comet experiment
tgaddair ecaf86d
Fixed callbacks
tgaddair aa4cb79
Fixed tests
tgaddair 932cb9c
Fixed Ray Tune callbacks
tgaddair 1e1dcad
Added ExceptionVariable
tgaddair 8662247
Fixed import order
tgaddair 5a517c8
Use LocalTestBackend
tgaddair b2fb225
Added LazyLoader
tgaddair 67dd0cb
Fixed tests
tgaddair 04db6cd
WIP mlflow
tgaddair 1e3a298
Added mlflow logging
tgaddair 5991eac
Reuse existing experiments
tgaddair 3b5cce0
Prepare Ray Tune
tgaddair d18e6ba
Use LazyLoader
tgaddair e50362e
Flatten and chunk params
tgaddair 6279b16
Log full config file
tgaddair 7ce8770
Fixed config and artifact logging
tgaddair 509cbc0
Added hparam
tgaddair e5185ba
Added tests
tgaddair 3d0da62
Added hyperopt test
tgaddair 0c8f07c
WIP mlflow model registry
tgaddair ff2ac86
Added export to mlflow
tgaddair 6259e3d
Log model artifacts
tgaddair cd16da1
Log model at the end of each epoch
tgaddair 74dd62c
Added model to test
tgaddair cf02d79
Removed unused import
tgaddair 9d84169
Merge master
tgaddair 75e17c5
model_name -> registered_model_name
tgaddair 39d0a1f
Removed redundant test
tgaddair File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Note: this whole block of changes is just adding a
try ... finally
to ensureon_train_end
is also called in the event of an exception.