Z-score Threshold tutorial (#1009)

addition of zscore mct tutorial including updates to tutorial utils
sony · Mar 27, 2024 · b773f42 · b773f42
1 parent f2c58cf
commit b773f42
Show file tree

Hide file tree

Showing 3 changed files with 987 additions and 91 deletions.
diff --git a/tutorials/notebooks/keras/ptq/example_keras_activation_threshold_search.ipynb b/tutorials/notebooks/keras/ptq/example_keras_activation_threshold_search.ipynb
@@ -267,10 +267,13 @@
       },
       "outputs": [],
       "source": [
-        "representative_dataset_folder = './imagenet/val'\n",
-        "batch_size = 20\n",
+        "REPRESENTATIVE_DATASET_FOLDER = './imagenet/val'\n",
+        "BATCH_SIZE = 20\n",
         "fraction =0.001\n",
-        "representative_dataset_gen = tutorial_tools.get_representative_dataset(fraction, representative_dataset_folder, batch_size)"
+        "model_version = 'MobileNetV2'\n",
+        "\n",
+        "preprocessor = tutorial_tools.DatasetPreprocessor(model_version=model_version)\n",
+        "representative_dataset_gen = preprocessor.get_representative_dataset(fraction, REPRESENTATIVE_DATASET_FOLDER, BATCH_SIZE)"
       ]
     },
     {