ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-05 01:25:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

ML - Added model management dialog to edit and delete models (#10125)

Browse Source 
* ML - Added model management dialog to edit and delete models
This commit is contained in:
Leila Lali
2020-04-24 08:33:35 -07:00
committed by GitHub
parent 7633c810aa
commit 1e8a9c47cb
34 changed files with 1184 additions and 462 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extensions/machine-learning-services/src/prediction/predictService.ts
View File

@@ -8,7 +8,7 @@ import * as azdata from 'azdata';
import { ApiWrapper } from '../common/apiWrapper';
import { QueryRunner } from '../common/queryRunner';
import * as utils from '../common/utils';
import { RegisteredModel } from '../modelManagement/interfaces';
import { ImportedModel } from '../modelManagement/interfaces';
import { PredictParameters, PredictColumn, DatabaseTable, TableColumn } from '../prediction/interfaces';
/**
@@ -42,7 +42,7 @@ export class PredictService {
*/
public async generatePredictScript(
predictParams: PredictParameters,
registeredModel: RegisteredModel | undefined,
registeredModel: ImportedModel | undefined,
filePath: string | undefined
): Promise<string> {
let connection = await this.getCurrentConnection();
@@ -146,9 +146,9 @@ WHERE TABLE_TYPE = 'BASE TABLE' AND TABLE_CATALOG='${utils.doubleEscapeSingleQuo
const threePartTableName = utils.getRegisteredModelsThreePartsName(importTable.databaseName || '', importTable.tableName || '', importTable.schema || '');
return `
DECLARE @model VARBINARY(max) = (
SELECT artifact_content
SELECT model
FROM ${threePartTableName}
WHERE artifact_id = ${modelId}
WHERE model_id = ${modelId}
);
WITH predict_input
AS (