How is it possible to complete the dataset column in MLflow?

Question

As mentioned in the title, I want to complete the dataset column but I don't know how.. I'm new to MLflow. I have a small python application in which there is a dog and cat classification.

Screen with the dataset column

I tried some python line but I didn't find..

Answer 1

See:

• https://mlflow.org/docs/latest/python_api/mlflow.html?highlight=log_input#mlflow.log_input
• https://mlflow.org/docs/latest/python_api/mlflow.data.html#mlflow.data.dataset.Dataset

Example:

dataset = mlflow.data.from_pandas(X_train,
  source="wine_quality_white.csv",
  name="wine_quality_white"
)
mlflow.log_input(dataset, context="training")