MLFlowResultTracker¶
- class MLFlowResultTracker(tracking_uri=None, experiment_id=None, experiment_name=None)[source]¶
Bases:
pykeen.trackers.ResultTracker
A tracker for MLFlow.
Initialize result tracking via MLFlow.
- Parameters
experiment_id (
Optional
[int
]) – The experiment ID. If given, this has to be the ID of an existing experiment in MFLow. Has priority over experiment_name.experiment_name (
Optional
[str
]) – The experiment name. If this experiment name exists, add the current run to this experiment. Otherwise create an experiment of the given name.
Methods Summary
end_run
()End a run.
log_metrics
(metrics[, step, prefix])Log metrics to result store.
log_params
(params[, prefix])Log parameters to result store.
start_run
([run_name])Start a run with an optional name.
Methods Documentation