feat: Update Tileio demo doc.

AmbiqAI · Sep 24, 2024 · 373c8fe · 373c8fe
1 parent 4606503
commit 373c8fe
Show file tree

Hide file tree

Showing 12 changed files with 11 additions and 122 deletions.
diff --git a/docs/assets/guides/hk-dashboard.jpeg b/docs/assets/guides/hk-dashboard.jpeg
diff --git a/docs/assets/heartkit-banner.png b/docs/assets/heartkit-banner.png
diff --git a/docs/guides/heartkit-demo.md b/docs/guides/heartkit-demo.md
diff --git a/docs/guides/index.md b/docs/guides/index.md
@@ -17,4 +17,4 @@ This section contains guides to help with various aspects of HeartKit. The guide
 ## <span class="sk-h2-span">Hardware Guides</span>
 
 - **[Run simple demo on EVB]()**: Running a demo using Ambiq SoC as backend inference engine.
-- **[Full HeartKit EVB App](heartkit-demo.md)**: A guide to running a multi-headed model demo on Ambiq EVB.
+- **[HeartKit Tileio Demo](https://ambiqai.github.io/tileio-docs/demos/heartkit/)**: A guide to running a multi-headed model demo on Ambiq EVB.
diff --git a/docs/tasks/denoise.md b/docs/tasks/denoise.md
@@ -44,7 +44,7 @@ Dataloaders are available for the following datasets:
 
 ## <span class="sk-h2-span">Pre-trained Models</span>
 
-The following table provides the latest performance and accuracy results of denoising models. Additional result details can be found in [Model Zoo → Denoise](../zoo/denoise.md).
+The following table provides the latest performance and accuracy results of denoising models. Additional result details can be found in [Model Zoo → Denoise](../zoo/index.md).
 
 
 --8<-- "assets/zoo/denoise/denoise-model-zoo-table.md"

diff --git a/docs/tasks/rhythm.md b/docs/tasks/rhythm.md
@@ -77,7 +77,7 @@ Dataloaders are available for the following datasets:
 
 ## <span class="sk-h2-span">Pre-trained Models</span>
 
-The following table provides the latest performance and accuracy results for rhythm models. Additional result details can be found in [Model Zoo → Rhythm](../zoo/rhythm.md).
+The following table provides the latest performance and accuracy results for rhythm models. Additional result details can be found in [Model Zoo → Rhythm](../zoo/index.md).
 
 
 --8<-- "assets/zoo/rhythm/rhythm-model-zoo-table.md"

diff --git a/docs/tasks/segmentation.md b/docs/tasks/segmentation.md
@@ -56,7 +56,7 @@ Dataloaders are available for the following datasets:
 
 ## <span class="sk-h2-span">Pre-Trained Models</span>
 
-The following table provides the latest performance and accuracy results for segmentation models. Additional result details can be found in [Model Zoo → Segmentation](../zoo/segmentation.md).
+The following table provides the latest performance and accuracy results for segmentation models. Additional result details can be found in [Model Zoo → Segmentation](../zoo/index.md).
 
 --8<-- "assets/zoo/segmentation/segmentation-model-zoo-table.md"
 

diff --git a/docs/zoo/index.md b/docs/zoo/index.md
@@ -13,7 +13,7 @@ The following table provides the latest performance and accuracy results for den
 | __DEN-PPG-TCN-SM__  | Synthetic         | 100Hz | 2.5s     | TCN            | 3.5K   | 1.1M    | 92.1% COS   |
 
 
-## <span class="sk-h2-span">[Signal Segmentation Task](./tasks/segmentation.md)</span>
+## <span class="sk-h2-span">[Signal Segmentation Task](../tasks/segmentation.md)</span>
 
 The following table provides the latest performance and accuracy results for ECG segmentation models.
 
@@ -25,7 +25,7 @@ The following table provides the latest performance and accuracy results for ECG
 | __SEG-PPG-2-TCN-SM__ | Synthetic                | 100Hz | 2.5s     | 2         | TCN           | 4K     | 1.43M   | 98.6% F1  |
 
 
-## <span class="sk-h2-span">[Rhythm Classification Task](./tasks/rhythm.md)</span>
+## <span class="sk-h2-span">[Rhythm Classification Task](../tasks/rhythm.md)</span>
 
 The following table provides the latest performance and accuracy results for rhythm classification models.
 
@@ -35,7 +35,7 @@ The following table provides the latest performance and accuracy results for rhy
 | __ARR-4-EFF-SM__ | LSAD                     | 100Hz | 5s       | 4         | EfficientNetV2 | 27K    |  1.6M   | 95.9% F1 |
 
 
-## <span class="sk-h2-span">[Beat Classification Task](./tasks/beat.md)</span>
+## <span class="sk-h2-span">[Beat Classification Task](../tasks/beat.md)</span>
 
 The following table provides the latest performance and accuracy results for beat classification models.
 

diff --git a/heartkit/datasets/lsad.py b/heartkit/datasets/lsad.py
@@ -481,7 +481,7 @@ def filter_patients_for_labels(
         patients_labels = self.get_patients_labels(patient_ids, label_map, label_type)
         # Find any patient with empty list
         label_mask = np.array([len(x) > 0 for x in patients_labels])
-        neg_mask = label_mask == -1
+        neg_mask = ~label_mask
         num_neg = neg_mask.sum()
         if num_neg > 0:
             logger.debug(f"Removed {num_neg} of {patient_ids.size} patients w/ no target class")

diff --git a/heartkit/datasets/ptbxl.py b/heartkit/datasets/ptbxl.py
@@ -565,7 +565,7 @@ def filter_patients_for_labels(
         patients_labels = self.get_patients_labels(patient_ids, label_map, label_type)
         # Find any patient with empty list
         label_mask = np.array([len(x) > 0 for x in patients_labels])
-        neg_mask = label_mask == -1
+        neg_mask = ~label_mask
         num_neg = neg_mask.sum()
         if num_neg > 0:
             logger.debug(f"Removed {num_neg} of {patient_ids.size} patients w/ no target class")

diff --git a/heartkit/tasks/denoise/datasets.py b/heartkit/tasks/denoise/datasets.py
@@ -43,8 +43,8 @@ def create_data_pipeline(
             drop_remainder=True,
             num_parallel_calls=tf.data.AUTOTUNE,
         )
-    ds = ds.map(lambda x: preprocessor(x), num_parallel_calls=tf.data.AUTOTUNE)
     ds = ds.map(lambda x: (augmenter(x), x), num_parallel_calls=tf.data.AUTOTUNE)
+    ds = ds.map(lambda x, y: (preprocessor(x), preprocessor(y)), num_parallel_calls=tf.data.AUTOTUNE)
     return ds.prefetch(tf.data.AUTOTUNE)
 
 

diff --git a/mkdocs.yml b/mkdocs.yml
@@ -101,7 +101,7 @@ nav:
     - Hardware Guides:
       - EVB Setup: guides/evb-setup.md
       - Rhythm Demo: guides/rhythm-demo.md
-      - HeartKit Demo: guides/heartkit-demo.md
+      - HeartKit Tileio Demo →: https://ambiqai.github.io/tileio-docs/demos/heartkit/
 
   - API: api/