扩展MLflow的功能
mlflow-extend的Python项目详细描述
MLflow扩展
扩展MLflow的功能。在
安装
来自PyPI
pip install mlflow-extend
来自GitHub(开发版本)
^{pr2}$示例
importnumpyasnpimportpandasaspdimportmatplotlib.pyplotaspltfromplotlyimportgraph_objectsasgofrommlflow_extendimportmlflowwithmlflow.start_run():# mlflow native APIsmlflow.log_param('param',0)mlflow.log_metric('metric',1.0)##### new APIs mlflow_extend provides ###### flatten dictmlflow.log_params_flatten({"a":{"b":0}})mlflow.log_metrics_flatten({"a":{"b":0.0}})# dictmlflow.log_dict({'a':0},'dict.json')# numpy arraymlflow.log_numpy(np.array([0]),'array.npy')# pandas dataframemlflow.log_df(pd.DataFrame({'a':[0]}),'df.csv')# matplotlib figurefig,ax=plt.subplots()ax.plot([0,1],[0,1])mlflow.log_figure(fig,'figure.png')# plotly figurefig=go.Figure(data=[go.Bar(x=[1,2,3],y=[1,3,2])])mlflow.log_figure(fig,'figure.html')# confusion matrixmlflow.log_confusion_matrix([[1,2],[3,4]])# run "mlflow ui" and see the result.
- 项目
标签: