Merge pull request #21 from kkkkcccc88/main

TKAN IMAGE
remigenet · Dec 16, 2024 · b9419f5 · b9419f5
2 parents 8a1de0c + 2867cc8
commit b9419f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -10,7 +10,6 @@ In case of performance consideration, the best setup tested used [jax docker ima
 I also discourage using as is the example for torch, it seems that currently when running test using torch backend with keras is much slower than torch directly, even for GRU or LSTM. 
 
 ![TKAN representation](image/TKAN.drawio.png)
-
 ## Installation
 
 Install TKAN directly from PyPI:
@@ -45,6 +44,11 @@ model = keras.Sequential([
 
 You can find a more complete example with comparison with other models in the example folder.
 
+## Other visualizations
+
+I thank [kkkkcccc88](https://github.com/kkkkcccc88) for this participation to the project by creating this visualization of the TKAN architecture.
+![TKAN IMAGE](image/TKAN_IMAGE.png)
+
 Please cite our work if you use this repo:
 
 ```

diff --git a/image/TKAN_IMAGE.png b/image/TKAN_IMAGE.png