logging the resampling strategy in form of a tag

JuliaAI · Aug 17, 2023 · ea66035 · ea66035
1 parent ba42e6f
commit ea66035
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/base.jl b/src/base.jl
@@ -3,7 +3,12 @@ function log_evaluation(logger::MLFlowLogger, performance_evaluation)
  artifact_location=logger.artifact_location)
  model_name = name(performance_evaluation.model)
  run = createrun(logger.service, experiment;
- run_name="$(model_name) run")
+ run_name="$(model_name) run",
+ tags=[
+ Dict("key" => "resampling", "value" => string(performance_evaluation.resampling)),
+ Dict("key" => "repeats", "value" => string(performance_evaluation.repeats)),
+ ]
+ )
 
  logmodelparams(logger.service, run, performance_evaluation.model)
  logmachinemeasures(logger.service, run, performance_evaluation.measure,