Unverified Commit 85c0d841 authored by Lijiaoa's avatar Lijiaoa Committed by GitHub
Browse files

Support show customized tuner/assessor name and get optimize_mode from...

Support show customized tuner/assessor name and get optimize_mode from tuner/advisor/assessor (#3317)
parent fe974e6b
...@@ -11,26 +11,17 @@ export const Command1 = (): any => { ...@@ -11,26 +11,17 @@ export const Command1 = (): any => {
const builtinName: string[] = []; const builtinName: string[] = [];
if (tuner !== undefined) { if (tuner !== undefined) {
title.push('Tuner'); title.push('Tuner');
if (tuner.builtinTunerName !== undefined) { builtinName.push(tuner.builtinTunerName || tuner.className || 'unknown');
builtinName.push(tuner.builtinTunerName);
}
} }
if (advisor !== undefined) { if (advisor !== undefined) {
title.push('Advisor'); title.push('Advisor');
if (advisor.builtinAdvisorName !== undefined) { builtinName.push(advisor.builtinAdvisorName || advisor.className || 'unknown');
builtinName.push(advisor.builtinAdvisorName);
}
if (advisor.className !== undefined) {
builtinName.push(advisor.className);
}
} }
if (assessor !== undefined) { if (assessor !== undefined) {
title.push('Assessor'); title.push('Assessor');
if (assessor.builtinAssessorName !== undefined) { builtinName.push(assessor.builtinAssessorName || assessor.className || 'unknown');
builtinName.push(assessor.builtinAssessorName);
}
} }
return ( return (
......
...@@ -128,7 +128,19 @@ class Experiment { ...@@ -128,7 +128,19 @@ class Experiment {
get optimizeMode(): string { get optimizeMode(): string {
const tuner = this.profile.params.tuner; const tuner = this.profile.params.tuner;
return tuner && tuner.classArgs && tuner.classArgs.optimize_mode ? tuner.classArgs.optimize_mode : 'unknown'; const advisor = this.profile.params.advisor;
const assessor = this.profile.params.assessor;
const resultTuner =
tuner && tuner.classArgs && tuner.classArgs.optimize_mode ? tuner.classArgs.optimize_mode : undefined;
const resultAdvisor =
advisor && advisor.classArgs && advisor.classArgs.optimize_mode
? advisor.classArgs.optimize_mode
: undefined;
const resultAssessor =
assessor && assessor.classArgs && assessor.classArgs.optimize_mode
? assessor.classArgs.optimize_mode
: undefined;
return resultTuner || resultAdvisor || resultAssessor || 'unknown';
} }
get trainingServicePlatform(): string { get trainingServicePlatform(): string {
......
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