From: ashishj1729 Date: Tue, 24 Dec 2024 07:55:50 +0000 (+0530) Subject: Amending retrain-pipeline as per Model-deployment requirements X-Git-Tag: 4.0.0~13 X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=refs%2Fchanges%2F60%2F13960%2F1;p=portal%2Faiml-dashboard.git Amending retrain-pipeline as per Model-deployment requirements 1. Model-path is deduced by modelversion 2. Only retrain-model will be present in Model.zip Change-Id: I0bc49b876b1d298b8d558e48f4c32c49fde2f0bf Signed-off-by: ashishj1729 --- diff --git a/kf-pipelines/qoe-pipeline-retrain-2.ipynb b/kf-pipelines/qoe-pipeline-retrain-2.ipynb index f933c35..f0ab77d 100644 --- a/kf-pipelines/qoe-pipeline-retrain-2.ipynb +++ b/kf-pipelines/qoe-pipeline-retrain-2.ipynb @@ -131,10 +131,7 @@ "\n", " # Load the model in SavedModel format\n", " # Path to the directory containing the saved model\n", - " if artifactversion ==\"1.0.0\":\n", - " model_path = \"./Model/1\"\n", - " else:\n", - " model_path= \"./Model/1/retrain\"\n", + " model_path = f\"./Model/{modelversion}\"\n", " \n", " model = tf.keras.models.load_model(model_path)\n", " model.compile(loss='mse', optimizer='adam', metrics=['mse'])\n", @@ -199,7 +196,7 @@ " print(updated_model_info)\n", " \n", " mm_sdk.upload_metrics(data, modelname, modelversion,new_artifactversion)\n", - " mm_sdk.upload_model(\"./\", modelname, modelversion, new_artifactversion)\n" + " mm_sdk.upload_model(\"./retrain/\", modelname, modelversion, new_artifactversion)\n" ] }, {