Skip to content
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

Keras 3.0 Logging With MLflow #10415

Merged
merged 42 commits into from Dec 3, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
42 commits
Select commit Hold shift + click to select a range
52edbf9
keras + mlflow tutorial
jessechancy Nov 15, 2023
261d404
update keras docs to include text
jessechancy Nov 16, 2023
e7ebe3c
linter
jessechancy Nov 16, 2023
a43f8fb
update linter
jessechancy Nov 16, 2023
46d4fb9
fix circleci
jessechancy Nov 16, 2023
54a87d6
comment changes
jessechancy Nov 20, 2023
40e6c16
remove html
jessechancy Nov 20, 2023
cb9a39a
linter
jessechancy Nov 20, 2023
409256e
linter
jessechancy Nov 30, 2023
ee41764
linter
jessechancy Nov 30, 2023
ac98462
rerun tests
jessechancy Dec 1, 2023
cc600e4
update title levels
jessechancy Dec 1, 2023
9b4f66b
update doc images
jessechancy Dec 1, 2023
0281738
update images
jessechancy Dec 1, 2023
b911d18
notebook
jessechancy Dec 1, 2023
8c683d1
linter
jessechancy Dec 1, 2023
c051b9e
linter
jessechancy Dec 1, 2023
74a9e92
changed to vanilla jupyter
jessechancy Dec 1, 2023
10de4f4
no outputs
jessechancy Dec 1, 2023
1916452
no outputs
jessechancy Dec 1, 2023
94230f4
comment out latter half test
jessechancy Dec 1, 2023
aedf1a5
comment out latter half
jessechancy Dec 1, 2023
b166e3e
add mpl
jessechancy Dec 1, 2023
4827bca
mpl
jessechancy Dec 1, 2023
bb74491
add images and run
jessechancy Dec 1, 2023
079e3d3
removed images'
jessechancy Dec 1, 2023
86d7f8d
remove loader
jessechancy Dec 1, 2023
25e90e4
verbose=0
jessechancy Dec 1, 2023
3a1fa5f
linter
jessechancy Dec 1, 2023
e5caf10
remove all loaders
jessechancy Dec 1, 2023
992934c
remove
jessechancy Dec 1, 2023
1db4556
update loader
jessechancy Dec 1, 2023
729ef69
remove mlflow and fit
jessechancy Dec 1, 2023
f441155
add fit without mlflow
jessechancy Dec 1, 2023
108a26a
add all with print end
jessechancy Dec 1, 2023
7b5181e
change with statement
jessechancy Dec 1, 2023
208419f
add print
jessechancy Dec 1, 2023
96e5553
run with print
jessechancy Dec 1, 2023
f2209d3
remove mlflow output
jessechancy Dec 1, 2023
cd86d08
final version
jessechancy Dec 1, 2023
56f606a
add summary
jessechancy Dec 1, 2023
3a4e6ef
small change to python file to trigger tests
jessechancy Dec 2, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
24 changes: 24 additions & 0 deletions docs/source/deep-learning/index.rst
Expand Up @@ -74,7 +74,31 @@ The officially supported integrations for deep learning libraries in MLflow enco
</section>

Harness the power of these integrations and elevate your deep learning projects with MLflow's comprehensive support.
For detailed guide on how to integrate MLflow with these libraries, refer to the following pages:

.. raw:: html

<section>
<article class="simple-grid">
<div class="simple-card">
<a href="keras/index.html">
<div class="header">
Keras
</div>
<p>
Learn about MLflow's native integration with the Keras library and see example notebooks that leverage
MLflow and Keras to build deep learning workflows.
</p>
</a>
</div>
</article>
</section>

.. toctree::
:maxdepth: 1
:hidden:

keras/index

MLflow Tracking for Deep Learning
---------------------------------
Expand Down
39 changes: 39 additions & 0 deletions docs/source/deep-learning/keras/index.rst
@@ -0,0 +1,39 @@
MLflow Keras 3.0 Integration
============================

Introduction
------------

Keras is a deep learning API written in Python, running on top of the machine learning platform TensorFlow.
It was developed with a focus on enabling fast experimentation.

Keras 3.0 (Keras Core) makes it possible to run Keras workflows on top of TensorFlow, JAX, and PyTorch.
It also enables you to seamlessly integrate Keras components (like layers, models, or metrics) as part of
low-level TensorFlow, JAX, and PyTorch workflows.

MLflow provides built-in support for Keras 3.0 workflows. It provides a callback that allows you to
log parameters and metrics during model training. Model logging is not currently supported.

5 Minute Quick Start with MLflow + Keras 3.0
--------------------------------------------

To get a quick overview of how to use MLflow + Keras 3.0, please read the quickstart guide. It will walk
you through how to use the callback for tracking experiments, as well as how to customize it.

.. raw:: html

<a href="quickstart/quickstart_keras_core.html" class="download-btn">View the Quickstart</a>

To download the Keras 3.0 tutorial notebook to run in your environment, click the link below:

.. raw:: html

<a href="https://raw.githubusercontent.com/mlflow/mlflow/master/docs/source/deep-learning/keras/quickstart/quickstart_keras_core.ipynb"
class="notebook-download-btn">Download the Quickstart of MLflow Keras Integration</a><br>


.. toctree::
:maxdepth: 1
:hidden:

quickstart/quickstart_keras_core.ipynb
526 changes: 526 additions & 0 deletions docs/source/deep-learning/keras/quickstart/quickstart_keras_core.ipynb

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion mlflow/sklearn/__init__.py
Expand Up @@ -165,7 +165,7 @@ def save_model(
metadata=None,
):
"""
Save a scikit-learn model to a path on the local file system. Produces an MLflow Model
Save a scikit-learn model to a path on the local file system. Produces a MLflow Model
containing the following flavors:

- :py:mod:`mlflow.sklearn`
Expand Down