Commit aedd1e5b authored by Svetlana Pavlitskaya's avatar Svetlana Pavlitskaya
Browse files

added EvalMetric to template

parent 2ff77c84
Pipeline #63452 passed with stages
in 1 minute and 51 seconds
......@@ -111,6 +111,13 @@ public class CNNTrainTemplateController {
return getConfiguration().getEntry("context").getValue().toString();
}
public String getEvalMetric() {
if (!getConfiguration().getEntryMap().containsKey("eval_metric")) {
return null;
}
return getConfiguration().getEntry("eval_metric").getValue().toString();
}
public String getOptimizerName() {
if (getConfiguration().getOptimizer() == null) {
return null;
......
......@@ -13,6 +13,9 @@ context = '${tc.context}',
<#if (tc.normalize)??>
normalize = ${tc.normalize?string("True","False")},
</#if>
<#if (tc.evalMetric)??>
eval_metric = '${tc.evalMetric}',
</#if>
<#if (tc.configuration.optimizer)??>
optimizer = '${tc.optimizerName}',
optimizer_params = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment