andras/bipolaroid
Watch
1
0
Fork
You've already forked bipolaroid
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

minor improvements

Browse source
This commit is contained in:
Andras Schmelczer 2024-04-28 21:26:46 +01:00
commit 703019c4cd
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 123 additions and 214 deletions
Download patch file Download diff file Expand all files Collapse all files

11
editor/utils/kldiv.py Normal file
View file

@ -0,0 +1,11 @@
import numpy as np
def kldiv(P: np.ndarray, Q: np.ndarray) -> float:
P /= P.sum()
Q /= Q.sum()
P_safe = np.maximum(P, np.finfo(float).eps)
Q_safe = np.maximum(Q, np.finfo(float).eps)
return np.sum(P_safe * np.log(P_safe / Q_safe))