Extend MLflow API
Project description
MLflow Extend
Extend MLflow's functionality.
Examples
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from mlflow_extend import mlflow
with mlflow.start_run():
# dict
mlflow.log_dict({'a', 0}, 'dict.json')
# numpy array
mlflow.log_numpy(np.array([0]), 'array.npy')
# pandas dataframe
mlflow.log_df(pd.DataFrame({'a': [0]}), 'df.csv')
# matplotlib figure
fig, ax = plt.subplots()
ax.plot([0, 1], [0, 1])
mlflow.log_figure(fig, 'figure.png')
Lint
# Run lint checking with black and flake8.
./dev/lint.sh
Test
# Run all the tests.
./dev/test.sh
# Save figures generated during the tests to .pytest_basetemp.
./dev/test.sh --savefig
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mlflow-extend-0.1.0.tar.gz
(4.0 kB
view hashes)
Built Distribution
Close
Hashes for mlflow_extend-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 52784d47900eb3bee1a63d51c770bd26eb5361242b85c2453a1fd5745aa6b8a5 |
|
MD5 | f4158cb4ed213fbcf747c8aa2ac3fd09 |
|
BLAKE2b-256 | 8489a4dbd7c46a225b632af376589a2be85c33608513be9b48d4d8b72c6cf855 |