From 1011f68c3eafb8b468cf1df23117b98c7c7b55ed Mon Sep 17 00:00:00 2001
From: gdamms <damguillotin@gmail.com>
Date: Tue, 25 Jun 2024 09:31:24 +0200
Subject: implement ation JS divergence

---
 main.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'main.py')

diff --git a/main.py b/main.py
index 81f832d..7c7daa1 100644
--- a/main.py
+++ b/main.py
@@ -423,5 +423,11 @@ if __name__ == '__main__':
         kl_score = kl(reals, fakes)
         print(f"KL divergence: {kl_score}")
 
+        rkl_score = kl(fakes, reals)
+        print(f"Reverse KL divergence: {rkl_score}")
+
+        jsd_score = jsd(reals, fakes)
+        print(f"Jensen-Shannon divergence: {jsd_score}")
+
 
         # plt.show()
-- 
cgit v1.3.1