MlflowClient (MLflow Tracking API 0.8.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
`long`	`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(long experimentId)` Creates a new run under the given experiment with no application name.
`Service.RunInfo`	`createRun(long experimentId, String appName)` Creates a new run under the given experiment with the given application name.
`Service.RunInfo`	`createRun(Service.CreateRun request)` Creates a new run.
`void`	`deleteExperiment(long 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(long experimentId)`
`Optional<Service.Experiment>`	`getExperimentByName(String experimentName)`
`Service.Run`	`getRun(String runUuid)`
`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(long 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`	`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(long experimentId, String newName)` Update an experiment's name.
`void`	`restoreExperiment(long experimentId)` Restore an experiment marked for deletion.
`void`	`restoreRun(String runId)` Restores a deleted run with the given ID.
`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)
```
    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(long experimentId)
```
    Creates a new run under the given experiment with no application name.
    
    Returns:
    
    RunInfo created by the server
  - createRun
```
public Service.RunInfo createRun(long 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(long experimentId)
```
    Returns:
    
    a list of all RunInfos associated with the given experiment.
  - listExperiments
```
public List<Service.Experiment> listExperiments()
```
    Returns:
    
    a list of all Experiments.
  - getExperiment
```
public Service.GetExperiment.Response getExperiment(long 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 long 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(long experimentId)
```
    Mark an experiment and associated runs, params, metrics, etc for deletion.
  - restoreExperiment
```
public void restoreExperiment(long experimentId)
```
    Restore an experiment marked for deletion.
  - renameExperiment
```
public void renameExperiment(long 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.
  - 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

listExperiments

getExperiment

getExperimentByName

createExperiment

deleteExperiment

restoreExperiment

renameExperiment

deleteRun

restoreRun

logParam

logMetric

setTag

setTerminated

setTerminated

setTerminated

sendGet

sendPost

logArtifact

logArtifact

logArtifacts

logArtifacts

listArtifacts

listArtifacts

downloadArtifacts

downloadArtifacts