implement ation JS divergence

1 files changed, 6 insertions, 0 deletions

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()