kedro catalog yaml with mlflow

catalog.yaml

# in <root>/conf/base/catalog.yaml
 
insurance:
  type: pandas.CSVDataSet
  filepath: data/01_raw/train.csv
  layer: raw
 
model_input_table:
  type: pandas.ParquetDataSet
  filepath: data/03_primary/model_input_table.pq
  layer: primary
 
xgboost_pipe.clf:
  type: kedro_mlflow.io.artifacts.MlflowArtifactDataSet
  data_set:
    type: kedro_mlflow.io.models.MlflowModelSaverDataSet
    flavor: mlflow.sklearn
    filepath: data/06_models/xgboost_clf.pickle
 
# metrics for mlflow
xgboost_pipe.model_metrics:
  type: kedro_mlflow.io.metrics.MlflowMetricsDataSet
  prefix: metrics
 
# objects to upload in mlflow
xgboost_pipe.roc_graph:
  type: kedro_mlflow.io.artifacts.MlflowArtifactDataSet
  data_set:
    type: kedro.extras.datasets.matplotlib.MatplotlibWriter
    filepath: data/06_models/xgb.roc_plot.png