ai.databand.mlflow

Class DbndMlflowClient

java.lang.Object

org.mlflow.tracking.MlflowClient

ai.databand.mlflow.DbndMlflowClient

public class DbndMlflowClient
extends org.mlflow.tracking.MlflowClient

Field Summary

Fields inherited from class org.mlflow.tracking.MlflowClient

DEFAULT_EXPERIMENT_ID

Constructor Summary

Constructors

Constructor and Description
DbndMlflowClient(org.mlflow.tracking.MlflowClient mlflowClient, java.lang.String dbndTrackingUri)
DbndMlflowClient(org.mlflow.tracking.MlflowClient mlflowClient, java.lang.String dbndTrackingUri, ai.databand.config.DbndConfig config)

Method Summary

Modifier and Type	Method and Description
org.mlflow.api.proto.Service.RunInfo	createRun(org.mlflow.api.proto.Service.CreateRun request)
ai.databand.DbndApi	dbndApi()
void	logMetric(java.lang.String runId, java.lang.String key, double value, long timestamp, long step)
static org.mlflow.tracking.MlflowClient	newClient()
void	setTerminated(java.lang.String runId) We should complete task run attempt as well as task run here.
protected java.lang.String	sha1long(java.lang.String namespace, java.lang.String name)
protected java.lang.String	sha1short(java.lang.String namespace, java.lang.String name)
protected java.lang.String	uuid5(java.lang.String namespace, java.lang.String name)

Methods inherited from class org.mlflow.tracking.MlflowClient

createExperiment, createRun, createRun, deleteExperiment, deleteRun, deleteTag, downloadArtifacts, downloadArtifacts, getExperiment, getExperimentByName, getMetricHistory, getRun, listArtifacts, listArtifacts, listExperiments, listRunInfos, logArtifact, logArtifact, logArtifacts, logArtifacts, logBatch, logMetric, logParam, renameExperiment, restoreExperiment, restoreRun, searchRuns, searchRuns, searchRuns, searchRuns, searchRuns, sendGet, sendPost, setExperimentTag, setTag, setTerminated, setTerminated

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DbndMlflowClient

public DbndMlflowClient(org.mlflow.tracking.MlflowClient mlflowClient,
                        java.lang.String dbndTrackingUri)

DbndMlflowClient

public DbndMlflowClient(org.mlflow.tracking.MlflowClient mlflowClient,
                        java.lang.String dbndTrackingUri,
                        ai.databand.config.DbndConfig config)

Method Detail

newClient

public static org.mlflow.tracking.MlflowClient newClient()

dbndApi

public ai.databand.DbndApi dbndApi()

createRun

public org.mlflow.api.proto.Service.RunInfo createRun(org.mlflow.api.proto.Service.CreateRun request)

Overrides:

createRun in class org.mlflow.tracking.MlflowClient

setTerminated

public void setTerminated(java.lang.String runId)

We should complete task run attempt as well as task run here.

Overrides:

setTerminated in class org.mlflow.tracking.MlflowClient

Parameters:

runId - MlFlow run id

logMetric

public void logMetric(java.lang.String runId,
                      java.lang.String key,
                      double value,
                      long timestamp,
                      long step)

Overrides:

logMetric in class org.mlflow.tracking.MlflowClient

sha1short

protected java.lang.String sha1short(java.lang.String namespace,
                                     java.lang.String name)

sha1long

protected java.lang.String sha1long(java.lang.String namespace,
                                    java.lang.String name)

uuid5

protected java.lang.String uuid5(java.lang.String namespace,
                                 java.lang.String name)