diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/optimization.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/optimization.py b/python/optimization.py index 6eda84fa..2651760e 100644 --- a/python/optimization.py +++ b/python/optimization.py @@ -58,6 +58,7 @@ def KLfit(P, q, b): if __name__=="__main__": #write some small test - x = np.array([5, 4, 3, 2, 1]) + P = np.array([[5, 4, 3, 2, 1],[3, 3, 4, 3, 3]]) w = np.ones(5)/5 - test = KLfit(x, w, 2.5) + b = np.array([2.5, 3.3]) + test = KLfit(P, w, b) |