MlflowClient (MLflow Tracking API 0.9.1 API)

java.lang.Object
- org.mlflow.tracking.MlflowClient

public class MlflowClient
extends Object

Client to an MLflow Tracking Sever.

Constructor Summary

Constructors
Constructor and Description
`MlflowClient()` Returns a default client based on the MLFLOW_TRACKING_URI environment variable.
`MlflowClient(MlflowHostCredsProvider hostCredsProvider)` Creates a new MlflowClient; users should prefer constructing ApiClients via `MlflowClient()` or `MlflowClient(String)` if possible.
`MlflowClient(String trackingUri)` Instantiates a new client using the provided tracking uri.

Method Summary

All Methods Instance Methods Concrete Methods
Modifier and Type	Method and Description
`String`	`createExperiment(String experimentName)` Creates a new experiment using the default artifact location provided by the server.
`Service.RunInfo`	`createRun()` Creates a new run under the default experiment with no application name.
`Service.RunInfo`	`createRun(Service.CreateRun request)` Creates a new run.
`Service.RunInfo`	`createRun(String experimentId)` Creates a new run under the given experiment with no application name.
`Service.RunInfo`	`createRun(String experimentId, String appName)` Creates a new run under the given experiment with the given application name.
`void`	`deleteExperiment(String experimentId)` Mark an experiment and associated runs, params, metrics, etc for deletion.
`void`	`deleteRun(String runId)` Deletes a run with the given ID.
`File`	`downloadArtifacts(String runId)` Returns a local directory containing all artifacts within the run's artifact directory.
`File`	`downloadArtifacts(String runId, String artifactPath)` Returns a local file or directory containing all artifacts within the given artifactPath within the run's root artifactDirectory.
`Service.GetExperiment.Response`	`getExperiment(String experimentId)`
`Optional<Service.Experiment>`	`getExperimentByName(String experimentName)`
`Service.Run`	`getRun(String runUuid)` Gets metadata, params, tags, and metrics for a run.
`List<Service.FileInfo>`	`listArtifacts(String runId)` Lists the artifacts immediately under the run's root artifact directory.
`List<Service.FileInfo>`	`listArtifacts(String runId, String artifactPath)` Lists the artifacts immediately under the given artifactPath within the run's root artifact directory.
`List<Service.Experiment>`	`listExperiments()`
`List<Service.RunInfo>`	`listRunInfos(String experimentId)`
`void`	`logArtifact(String runId, File localFile)` Uploads the given local file to the run's root artifact directory.
`void`	`logArtifact(String runId, File localFile, String artifactPath)` Uploads the given local file to an artifactPath within the run's root directory.
`void`	`logArtifacts(String runId, File localDir)` Uploads all files within the given local directory the run's root artifact directory.
`void`	`logArtifacts(String runId, File localDir, String artifactPath)` Uploads all files within the given local director an artifactPath within the run's root artifact directory.
`void`	`logBatch(String runUuid, Iterable<Service.Metric> metrics, Iterable<Service.Param> params, Iterable<Service.RunTag> tags)` Log multiple metrics, params, and/or tags against a given run (argument runUuid).
`void`	`logMetric(String runUuid, String key, double value)` Logs a new metric against the given run, as a key-value pair.
`void`	`logParam(String runUuid, String key, String value)` Logs a parameter against the given run, as a key-value pair.
`void`	`renameExperiment(String experimentId, String newName)` Update an experiment's name.
`void`	`restoreExperiment(String experimentId)` Restore an experiment marked for deletion.
`void`	`restoreRun(String runId)` Restores a deleted run with the given ID.
`List<Service.RunInfo>`	`searchRuns(List<String> experimentIds, String searchFilter)` Returns runs from provided list of experiments, that satisfy the search query.
`List<Service.RunInfo>`	`searchRuns(List<String> experimentIds, String searchFilter, Service.ViewType runViewType)` Returns runs from provided list of experiments, that satisfy the search query.
`String`	`sendGet(String path)` Send a GET to the following path, including query parameters.
`String`	`sendPost(String path, String json)` Send a POST to the following path, with a String-encoded JSON body.
`void`	`setTag(String runUuid, String key, String value)` Logs a new tag against the given run, as a key-value pair.
`void`	`setTerminated(String runUuid)` Sets the status of a run to be FINISHED at the current time.
`void`	`setTerminated(String runUuid, Service.RunStatus status)` Sets the status of a run to be completed at the current time.
`void`	`setTerminated(String runUuid, Service.RunStatus status, long endTime)` Sets the status of a run to be completed at the given endTime.

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

- Constructor Detail
  - MlflowClient
```
public MlflowClient()
```
    Returns a default client based on the MLFLOW_TRACKING_URI environment variable.
  - MlflowClient
```
public MlflowClient(String trackingUri)
```
    Instantiates a new client using the provided tracking uri.
  - MlflowClient
```
public MlflowClient(MlflowHostCredsProvider hostCredsProvider)
```
    Creates a new MlflowClient; users should prefer constructing ApiClients via MlflowClient() or MlflowClient(String) if possible.
- Method Detail
  - getRun
```
public Service.Run getRun(String runUuid)
```
    Gets metadata, params, tags, and metrics for a run. In the case where multiple metrics with the same key are logged for the run, returns only the value with the latest timestamp. If there are multiple values with the latest timestamp, returns the maximum of these values.
    
    Returns:
    
    Run associated with the id.
  - createRun
```
public Service.RunInfo createRun()
```
    Creates a new run under the default experiment with no application name.
    
    Returns:
    
    RunInfo created by the server
  - createRun
```
public Service.RunInfo createRun(String experimentId)
```
    Creates a new run under the given experiment with no application name.
    
    Returns:
    
    RunInfo created by the server
  - createRun
```
public Service.RunInfo createRun(String experimentId,
                                 String appName)
```
    Creates a new run under the given experiment with the given application name.
    
    Returns:
    
    RunInfo created by the server
  - createRun
```
public Service.RunInfo createRun(Service.CreateRun request)
```
    Creates a new run. This method allows providing all possible fields of CreateRun, and can be invoked as follows:
```
   import org.mlflow.api.proto.Service.CreateRun;
   CreateRun.Builder request = CreateRun.newBuilder();
   request.setExperimentId(experimentId);
   request.setSourceVersion("my-version");
   createRun(request.build());
   
```
    Returns:
    
    RunInfo created by the server
  - listRunInfos
```
public List<Service.RunInfo> listRunInfos(String experimentId)
```
    Returns:
    
    a list of all RunInfos associated with the given experiment.
  - searchRuns
```
public List<Service.RunInfo> searchRuns(List<String> experimentIds,
                                        String searchFilter)
```
    Returns runs from provided list of experiments, that satisfy the search query.
    
    Parameters:
    
    experimentIds - List of experiment IDs
    
    searchFilter - SQL compatible search query string. Format of this query string is similar to that specified on MLflow UI. Example : "params.model = 'LogisticRegression' and metrics.acc = 0.9"
    
    Returns:
    
    a list of all RunInfos that satisfy search filter.
  - searchRuns
```
public List<Service.RunInfo> searchRuns(List<String> experimentIds,
                                        String searchFilter,
                                        Service.ViewType runViewType)
```
    Returns runs from provided list of experiments, that satisfy the search query.
    
    Parameters:
    
    experimentIds - List of experiment IDs
    
    searchFilter - SQL compatible search query string. Format of this query string is similar to that specified on MLflow UI. Example : "params.model = 'LogisticRegression' and metrics.acc != 0.9"
    
    runViewType - ViewType for expected runs. One of (ACTIVE_ONLY, DELETED_ONLY, ALL) Defaults to ACTIVE_ONLY.
    
    Returns:
    
    a list of all RunInfos that satisfy search filter.
  - listExperiments
```
public List<Service.Experiment> listExperiments()
```
    Returns:
    
    a list of all Experiments.
  - getExperiment
```
public Service.GetExperiment.Response getExperiment(String experimentId)
```
    Returns:
    
    an experiment with the given id.
  - getExperimentByName
```
public Optional<Service.Experiment> getExperimentByName(String experimentName)
```
    Returns:
    
    the experiment associated with the given name or Optional.empty if none exists.
  - createExperiment
```
public String createExperiment(String experimentName)
```
    Creates a new experiment using the default artifact location provided by the server.
    
    Parameters:
    
    experimentName - Name of the experiment. This must be unique across all experiments.
    
    Returns:
    
    experiment id of the newly created experiment.
  - deleteExperiment
```
public void deleteExperiment(String experimentId)
```
    Mark an experiment and associated runs, params, metrics, etc for deletion.
  - restoreExperiment
```
public void restoreExperiment(String experimentId)
```
    Restore an experiment marked for deletion.
  - renameExperiment
```
public void renameExperiment(String experimentId,
                             String newName)
```
    Update an experiment's name. The new name must be unique.
  - deleteRun
```
public void deleteRun(String runId)
```
    Deletes a run with the given ID.
  - restoreRun
```
public void restoreRun(String runId)
```
    Restores a deleted run with the given ID.
  - logParam
```
public void logParam(String runUuid,
                     String key,
                     String value)
```
    Logs a parameter against the given run, as a key-value pair. This cannot be called against the same parameter key more than once.
  - logMetric
```
public void logMetric(String runUuid,
                      String key,
                      double value)
```
    Logs a new metric against the given run, as a key-value pair. New values for the same metric may be recorded over time, and are marked with a timestamp.
  - setTag
```
public void setTag(String runUuid,
                   String key,
                   String value)
```
    Logs a new tag against the given run, as a key-value pair.
  - logBatch
```
public void logBatch(String runUuid,
                     Iterable<Service.Metric> metrics,
                     Iterable<Service.Param> params,
                     Iterable<Service.RunTag> tags)
```
    Log multiple metrics, params, and/or tags against a given run (argument runUuid). Argument metrics, params, and tag iterables can be nulls.
  - setTerminated
```
public void setTerminated(String runUuid)
```
    Sets the status of a run to be FINISHED at the current time.
  - setTerminated
```
public void setTerminated(String runUuid,
                          Service.RunStatus status)
```
    Sets the status of a run to be completed at the current time.
  - setTerminated
```
public void setTerminated(String runUuid,
                          Service.RunStatus status,
                          long endTime)
```
    Sets the status of a run to be completed at the given endTime.
  - sendGet
```
public String sendGet(String path)
```
    Send a GET to the following path, including query parameters. This is mostly an internal API, but allows making lower-level or unsupported requests.
    
    Returns:
    
    JSON response from the server
  - sendPost
```
public String sendPost(String path,
                       String json)
```
    Send a POST to the following path, with a String-encoded JSON body. This is mostly an internal API, but allows making lower-level or unsupported requests.
    
    Returns:
    
    JSON response from the server
  - logArtifact
```
public void logArtifact(String runId,
                        File localFile)
```
    Uploads the given local file to the run's root artifact directory. For example,
```
   logArtifact(runId, "/my/localModel")
   listArtifacts(runId) // returns "localModel"
   
```
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    localFile - File to upload. Must exist, and must be a simple file (not a directory).
  - logArtifact
```
public void logArtifact(String runId,
                        File localFile,
                        String artifactPath)
```
    Uploads the given local file to an artifactPath within the run's root directory. For example,
```
   logArtifact(runId, "/my/localModel", "model")
   listArtifacts(runId, "model") // returns "model/localModel"
   
```
    (i.e., the localModel file is now available in model/localModel).
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    localFile - File to upload. Must exist, and must be a simple file (not a directory).
    
    artifactPath - Artifact path relative to the run's root directory. Should NOT start with a /.
  - logArtifacts
```
public void logArtifacts(String runId,
                         File localDir)
```
    Uploads all files within the given local directory the run's root artifact directory. For example, if /my/local/dir/ contains two files "file1" and "file2", then
```
   logArtifacts(runId, "/my/local/dir")
   listArtifacts(runId) // returns "file1" and "file2"
   
```
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    localDir - Directory to upload. Must exist, and must be a directory (not a simple file).
  - logArtifacts
```
public void logArtifacts(String runId,
                         File localDir,
                         String artifactPath)
```
    Uploads all files within the given local director an artifactPath within the run's root artifact directory. For example, if /my/local/dir/ contains two files "file1" and "file2", then
```
   logArtifacts(runId, "/my/local/dir", "model")
   listArtifacts(runId, "model") // returns "model/file1" and "model/file2"
   
```
    (i.e., the contents of the local directory are now available in model/).
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    localDir - Directory to upload. Must exist, and must be a directory (not a simple file).
    
    artifactPath - Artifact path relative to the run's root directory. Should NOT start with a /.
  - listArtifacts
```
public List<Service.FileInfo> listArtifacts(String runId)
```
    Lists the artifacts immediately under the run's root artifact directory. This does not recursively list; instead, it will return FileInfos with isDir=true where further listing may be done.
    
    Parameters:
    
    runId - Run ID of an existing MLflow run.
  - listArtifacts
```
public List<Service.FileInfo> listArtifacts(String runId,
                                            String artifactPath)
```
    Lists the artifacts immediately under the given artifactPath within the run's root artifact directory. This does not recursively list; instead, it will return FileInfos with isDir=true where further listing may be done.
    
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    artifactPath - Artifact path relative to the run's root directory. Should NOT start with a /.
  - downloadArtifacts
```
public File downloadArtifacts(String runId)
```
    Returns a local directory containing *all* artifacts within the run's artifact directory. Note that this will download the entire directory path, and so may be expensive if the directory has a lot of data.
    
    Parameters:
    
    runId - Run ID of an existing MLflow run.
  - downloadArtifacts
```
public File downloadArtifacts(String runId,
                              String artifactPath)
```
    Returns a local file or directory containing all artifacts within the given artifactPath within the run's root artifactDirectory. For example, if "model/file1" and "model/file2" exist within the artifact directory, then
```
   downloadArtifacts(runId, "model") // returns a local directory containing "file1" and "file2"
   downloadArtifacts(runId, "model/file1") // returns a local *file* with the contents of file1.
   
```
    Note that this will download the entire subdirectory path, and so may be expensive if the subdirectory has a lot of data.
    Parameters:
    
    runId - Run ID of an existing MLflow run.
    
    artifactPath - Artifact path relative to the run's root directory. Should NOT start with a /.

Class MlflowClient

Constructor Summary

Method Summary

Methods inherited from class java.lang.Object

Constructor Detail

MlflowClient

MlflowClient

MlflowClient

Method Detail

getRun

createRun

createRun

createRun

createRun

listRunInfos

searchRuns

searchRuns

listExperiments

getExperiment

getExperimentByName

createExperiment

deleteExperiment

restoreExperiment

renameExperiment

deleteRun

restoreRun

logParam

logMetric

setTag

logBatch

setTerminated

setTerminated

setTerminated

sendGet

sendPost

logArtifact

logArtifact

logArtifacts

logArtifacts

listArtifacts

listArtifacts

downloadArtifacts

downloadArtifacts