From 5a65ecb5a88a2685cb6da633de4bdfb89fe91289 Mon Sep 17 00:00:00 2001 From: Avi Asulin Date: Tue, 21 May 2024 17:32:32 +0300 Subject: [PATCH] [fraud demo] update feature selection function --- fraud-prevention-feature-store/02-create-training-model.ipynb | 3 +++ 1 file changed, 3 insertions(+) diff --git a/fraud-prevention-feature-store/02-create-training-model.ipynb b/fraud-prevention-feature-store/02-create-training-model.ipynb index a00d5398..0f330a83 100644 --- a/fraud-prevention-feature-store/02-create-training-model.ipynb +++ b/fraud-prevention-feature-store/02-create-training-model.ipynb @@ -975,6 +975,9 @@ "source": [ "feature_selection_fn = mlrun.import_function('hub://feature_selection')\n", "\n", + "feature_selection_fn.with_requirements(requirements=[\"matplotlib==3.8.4\"])\n", + "feature_selection_fn.deploy()\n", + "\n", "feature_selection_run = feature_selection_fn.run(\n", " params={\"k\": 18,\n", " \"min_votes\": 2,\n",