ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-16 18:46:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

ML - Added a link in models page to run predict on a model (#13124)

Browse Source 
* Added a link in models page to run predict on a model

* Updated the icons
This commit is contained in:
Leila Lali
2020-10-29 16:37:23 -07:00
committed by GitHub
parent e31d563f61
commit d450588e39
10 changed files with 121 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions/machine-learning/src/views/models/prediction/columnsSelectionPage.ts
View File

@@ -94,7 +94,6 @@ export class ColumnsSelectionPage extends ModelViewBase implements IPageView, ID
if (modelParameters && this.inputColumnsComponent && this.outputColumnsComponent) {
this.inputColumnsComponent.modelParameters = modelParameters;
this.outputColumnsComponent.modelParameters = modelParameters;
await this.inputColumnsComponent.refresh();
await this.outputColumnsComponent.refresh();
}
} catch (error) {

8
extensions/machine-learning/src/views/models/prediction/predictWizard.ts
View File

@@ -31,7 +31,8 @@ export class PredictWizard extends ModelViewBase {
constructor(
apiWrapper: ApiWrapper,
root: string,
parent?: ModelViewBase) {
parent?: ModelViewBase,
private _selectedModels?: ImportedModel[] | undefined) {
super(apiWrapper, root);
this._parentView = parent;
this.modelActionType = ModelActionType.Predict;
@@ -44,7 +45,7 @@ export class PredictWizard extends ModelViewBase {
this.modelSourceType = ModelSourceType.RegisteredModels;
this.modelSourcePage = new ModelSourcePage(this._apiWrapper, this, [ModelSourceType.RegisteredModels, ModelSourceType.Local, ModelSourceType.Azure]);
this.columnsSelectionPage = new ColumnsSelectionPage(this._apiWrapper, this);
this.modelBrowsePage = new ModelBrowsePage(this._apiWrapper, this, false);
this.modelBrowsePage = new ModelBrowsePage(this._apiWrapper, this, false, this._selectedModels);
this.wizardView = new WizardView(this._apiWrapper);
let wizard = this.wizardView.createWizard(constants.makePredictionTitle,
@@ -83,6 +84,9 @@ export class PredictWizard extends ModelViewBase {
});
await wizard.open();
if (this._selectedModels) {
await wizard.setCurrentPage(wizard.pages.length - 1);
}
}
private onLoading(): void {