def log_param(self, run_uuid, param):
_validate_run_id(run_uuid)
_validate_param_name(param.key)
run = self.get_run(run_uuid)
param_path = self._get_param_path(run.info.experiment_id, run_uuid,
param.key)
make_containing_dirs(param_path)
write_to(param_path, "%s\n" % param.value)
def log_metric(self, run_uuid, metric):
_validate_run_id(run_uuid)
_validate_metric_name(metric.key)
run = self.get_run(run_uuid)
metric_path = self._get_metric_path(run.info.experiment_id, run_uuid,
metric.key)
make_containing_dirs(metric_path)
append_to(metric_path, "%s %s\n" % (metric.timestamp, metric.value))
def log_param(self, run_id, param):
_validate_run_id(run_id)
_validate_param_name(param.key)
run = self.get_run(run_id)
check_run_is_active(run.info)
param_path = self._get_param_path(run.info.experiment_id, run_id, param.key)
make_containing_dirs(param_path)
write_to(param_path, self._writeable_value(param.value))
def set_tag(self, run_uuid, tag):
_validate_run_id(run_uuid)
_validate_tag_name(tag.key)
run = self.get_run(run_uuid)
tag_path = self._get_tag_path(run.info.experiment_id, run_uuid,
tag.key)
make_containing_dirs(tag_path)
write_to(tag_path, "%s\n" % tag.value)
def set_tag(self, run_id, tag):
_validate_run_id(run_id)
_validate_tag_name(tag.key)
run = self.get_run(run_id)
check_run_is_active(run.info)
tag_path = self._get_tag_path(run.info.experiment_id, run_id, tag.key)
make_containing_dirs(tag_path)
# Don't add trailing newline
write_to(tag_path, self._writeable_value(tag.value))
def _log_run_param(self, run_info, param):
param_path = self._get_param_path(run_info.experiment_id,
run_info.run_id, param.key)
writeable_param_value = self._writeable_value(param.value)
if os.path.exists(param_path):
self._validate_new_param_value(param_path=param_path,
param_key=param.key,
run_id=run_info.run_id,
new_value=writeable_param_value)
make_containing_dirs(param_path)
write_to(param_path, writeable_param_value)
def set_experiment_tag(self, experiment_id, tag):
"""
Set a tag for the specified experiment
:param experiment_id: String ID of the experiment
:param tag: ExperimentRunTag instance to log
"""
_validate_tag_name(tag.key)
experiment = self.get_experiment(experiment_id)
if experiment.lifecycle_stage != LifecycleStage.ACTIVE:
raise MlflowException(
"The experiment {} must be in the 'active'"
"lifecycle_stage to set tags".format(experiment.experiment_id),
error_code=databricks_pb2.INVALID_PARAMETER_VALUE)
tag_path = self._get_experiment_tag_path(experiment_id, tag.key)
make_containing_dirs(tag_path)
write_to(tag_path, self._writeable_value(tag.value))
def _set_run_tag(self, run_info, tag):
tag_path = self._get_tag_path(run_info.experiment_id, run_info.run_id,
tag.key)
make_containing_dirs(tag_path)
# Don't add trailing newline
write_to(tag_path, self._writeable_value(tag.value))
def _log_run_param(self, run_info, param):
param_path = self._get_param_path(run_info.experiment_id,
run_info.run_id, param.key)
make_containing_dirs(param_path)
write_to(param_path, self._writeable_value(param.value))
def _log_run_metric(self, run_info, metric):
metric_path = self._get_metric_path(run_info.experiment_id,
run_info.run_id, metric.key)
make_containing_dirs(metric_path)
append_to(metric_path,
"%s %s %s\n" % (metric.timestamp, metric.value, metric.step))
