From 21f44eda163a7b6615983844a563f343958a2328 Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Sat, 7 Dec 2024 13:55:38 +0530 Subject: [PATCH] changing pipeline for the modelName and ModelVersion and artifactversion and featurepath Change-Id: Iffd7b383ac18f96bb6eac873aebdd347cd95c208 Signed-off-by: rajdeep11 --- kf-pipelines/qoe-pipeline.ipynb | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/kf-pipelines/qoe-pipeline.ipynb b/kf-pipelines/qoe-pipeline.ipynb index 1cb2c88..378eff4 100644 --- a/kf-pipelines/qoe-pipeline.ipynb +++ b/kf-pipelines/qoe-pipeline.ipynb @@ -24,12 +24,12 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ "@component(base_image=BASE_IMAGE)\n", - "def train_export_model(trainingjobName: str, epochs: str, version: str):\n", + "def train_export_model(featurepath: str, epochs: str, modelname: str, modelversion:str, artifactversion:str):\n", " \n", " import tensorflow as tf\n", " from numpy import array\n", @@ -47,8 +47,8 @@ " \n", " fs_sdk = FeatureStoreSdk()\n", " mm_sdk = ModelMetricsSdk()\n", - " print(\"job name is: \", trainingjobName)\n", - " features = fs_sdk.get_features(trainingjobName, ['pdcpBytesDl','pdcpBytesUl'])\n", + " print(\"featurepath is: \", featurepath)\n", + " features = fs_sdk.get_features(featurepath, ['pdcpBytesDl','pdcpBytesUl'])\n", " print(\"Dataframe:\")\n", " print(features)\n", "\n", @@ -103,13 +103,15 @@ " data['metrics'] = []\n", " data['metrics'].append({'Accuracy': str(np.mean(np.absolute(np.asarray(xx)-np.asarray(yy))<5))})\n", " \n", - " mm_sdk.upload_metrics(data, trainingjobName, version)\n", - " mm_sdk.upload_model(\"./\", trainingjobName, version)\n" + " artifactversion=\"1.0.0\"\n", + " \n", + " mm_sdk.upload_metrics(data, modelname, modelversion,artifactversion)\n", + " mm_sdk.upload_model(\"./\", modelname, modelversion, artifactversion)\n" ] }, { "cell_type": "code", - "execution_count": 4, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ @@ -118,9 +120,9 @@ " description=\"qoe\",\n", ")\n", "def super_model_pipeline( \n", - " trainingjob_name: str, epochs: str, version: str):\n", + " featurepath: str, epochs: str, modelname: str, modelversion:str, artifactversion:str):\n", " \n", - " trainop=train_export_model(trainingjobName=trainingjob_name, epochs=epochs, version=version)\n", + " trainop=train_export_model(featurepath=featurepath, epochs=epochs, modelname=modelname, modelversion=modelversion, artifactversion=artifactversion)\n", " trainop.set_caching_options(False)\n", " kubernetes.set_image_pull_policy(trainop, \"IfNotPresent\")" ] -- 2.16.6