mlflow.entities

All entities returned by the MLflow REST API.

class mlflow.entities.Experiment(experiment_id, name, artifact_location)

Bases: mlflow.entities._mlflow_object._MLflowObject

Experiment object for Python client.

DEFAULT_EXPERIMENT_ID = 0

artifact_location

String corresponding to the root artifact URI for the experiment.

experiment_id

Integer ID of the experiment.

classmethod from_proto(proto)

name

String name of the experiment.

to_proto()

class mlflow.entities.FileInfo(path, is_dir, file_size)

Bases: mlflow.entities._mlflow_object._MLflowObject

Class exposing metadata about a file or directory.

file_size

Size of the file or directory. May be None if the current FileInfo is a directory.

classmethod from_proto(proto)

is_dir

Boolean – whether the current FileInfo corresponds to a directory.

path

String path of the file or directory.

to_proto()

class mlflow.entities.Metric(key, value, timestamp)

Bases: mlflow.entities._mlflow_object._MLflowObject

Metric object for python client.

classmethod from_proto(proto)

key

String key corresponding to the metric’s name.

timestamp

Metric timestamp as an integer (typically, milliseconds since the Unix epoch).

to_proto()

value

Float value of the metric.

class mlflow.entities.Param(key, value)

Bases: mlflow.entities._mlflow_object._MLflowObject

Param object for python client.

classmethod from_proto(proto)

key

String key corresponding to the param name.

to_proto()

value

String value of the param.

class mlflow.entities.Run(run_info, run_data)

Bases: mlflow.entities._mlflow_object._MLflowObject

Run object for python client.

data

return – mlflow.entities.RunData

classmethod from_dictionary(the_dict)

classmethod from_proto(proto)

info

return – mlflow.entities.RunInfo

to_dictionary()

to_proto()

class mlflow.entities.RunData(metrics=None, params=None)

Bases: mlflow.entities._mlflow_object._MLflowObject

Class exposing run data (metrics and parameters).

classmethod from_dictionary(the_dict)

classmethod from_proto(proto)

metrics

List of mlflow.entities.metric.Metric for the current run.

params

List of mlflow.entities.param.Param for the current run.

to_proto()

class mlflow.entities.RunInfo(run_uuid, experiment_id, name, source_type, source_name, entry_point_name, user_id, status, start_time, end_time, source_version, tags, artifact_uri=None)

Bases: mlflow.entities._mlflow_object._MLflowObject

Class containing metadata for a run.

artifact_uri

String – root artifact URI of the run.

copy_with_overrides(status, end_time)

Returns a copy the current RunInfo with certain attributes modified

end_time

End time of the run, in number of milliseconds since the UNIX epoch.

entry_point_name

String name of the entry point for the run.

experiment_id

Integer ID of the experiment for the current run.

classmethod from_dictionary(the_dict)

classmethod from_proto(proto)

name

String name of the run.

run_uuid

String containing run UUID.

source_name

String name of the source of the run (GitHub URI of the project corresponding to the run, etc).

source_type

Enum (one of the values in mlflow.entities.source_type.SourceType) describing the source of the run.

source_version

String – git commit of the code used for the run, if available.

start_time

Start time of the run, in number of milliseconds since the UNIX epoch.

status

Enum (one of the values in mlflow.entities.run_status.RunStatus) describing the status of the run.

tags

List of mlflow.entities.run_tag.RunTag for the run.

to_proto()

user_id

String ID of the user who initiated this run.

class mlflow.entities.RunStatus

Bases: object

Enum describing the status of a mlflow.entities.run.Run.

FAILED = 4

FINISHED = 3

RUNNING = 1

SCHEDULED = 2

static from_string(status_str)

static is_terminated(status)

static to_string(status)

class mlflow.entities.RunTag(key, value)

Bases: mlflow.entities._mlflow_object._MLflowObject

RunTag object for Python client.

classmethod from_proto(proto)

key

String name of the tag.

to_proto()

value

String value of the tag.

class mlflow.entities.SourceType

Bases: object

Enum describing the originating source of a mlflow.entities.run.Run.

JOB = 2

LOCAL = 4

NOTEBOOK = 1

PROJECT = 3

UNKNOWN = 5